Python 将数组附加到最后一列

Question

下面的数组是我的模型的结果，如何将其附加到我的数据框的最后一列

In[]  logireg.predict(X.head(5))
Out[] array([0, 0, 0, 1, 0], dtype=int64)

数据框数据:

age   job       month
33   blue       apr
56   admin      jun
37   tech       aug
76   retired    jun
56   service    may

预期输出

age   job       month   predict
33   blue       apr       0
56   admin      jun       0
37   tech       aug       0
76   retired    jun       1
56   service    may       0

使用for循环或zip函数？

Answer 1

您可以将其直接分配给数据框列。假设您的数据框在此称为 dataframe:

predictions = logireg.predict(X.head(5))
dataframe['predict'] = predictions