python - 在 pandas 中使用回归？

Question

我有一个这样的数据集:

height  age  gender
 140     10    male
 150     15    male
 160     20    male
 135     12    female
 170     16    female
 180     20    female

到目前为止，我已经像这样操作数据，预测变量或响应变量是高度(Y):

Y=   140           X= 1     10      1        β=   β0
     150              1     15      1             β1                                                                                           
     160              1     20      1             β2                                                
     135              1     12      0
     170              1     16      0
     180              1     20      0

其中X矩阵中的1和0分别是男性和女性，

我想在Python中执行这个功能

Y= β0 + β1 X + β2 X

β= (XT x) XT Y

其中 T 是转置..我不知道如何执行它，因为我不熟悉 python

Answer 1

如果您安装了 numpy，请尝试以下操作:

import numpy as np

#Create a list for Y
y = [140,150,160,135,170,180]

#Convert to numpy array
Y = np.asarray(y).reshape(6,1)

#Create a list for X
x = [[1,10,1],[1,15,1],[1,20,1],[1,12,1],[1,16,1],[1,20,1]]

#Convert to numpy array
X = np.asarray(x).reshape(6,3)

#Multiply using np.matmul for matrix multiplication
beta = np.matmul(np.matmul(np.matmul(X.T,X),X.T),Y)

否则，在 python 解释器上运行上述代码行之前，请使用 Python 包管理器安装 numpy。 pip 的等效命令是:

sudo pip install numpy