给定一个身高体重数据集，如何利用python计算数据集通过输入身高预测体重呢

Question

import pandas as pd
import numpy as np
p=pd.read_excel('D:\zip\zip.xlsx',sheet_name='Sheet1')
m=np.array(p)
height=m[:,0]
weight=m[:,1]

Answer 1

from scipy import optimize

def func(x, k, b):
  return k * x + b

k, b = optimize.curve_fit(func, height, weight)[0]

h = input("身高")
w = k * h + b 
print("体重为", w)