pandas 点积仅整数、切片 (`:`)、省略号 (`...`)、numpy.newaxis (`None`) 和整数或布尔数组是有效索引

Question

我有这个名为 xTest 的数据框和这个向量 x 两种类型都是numpy.ndarray x 长度为 197 xtest 的形状是 (25000, 197)。 我正在尝试通过获取xTest 和x 的点积来为xTest 生成一个称为预测的新列。

我试过了：

xTest["pred"] = np.dot(xTest,x)

和

xTest["pred"] = xTest.dot(x)

我不断收到此错误： IndexError: only integers, slices (:), ellipsis (...), numpy.newaxis (None) and integer or boolean arrays are valid indices.

xTest 已经是一个热编码。

更新： 感谢下面指出的好人 我能够像这样解决它

df = pd.DataFrame()
df["pred"] = np.dot(xTest,x)

Answer 1

因为xTest 是一个numpy 数组而不是dataframe，你不能使用字符串来索引它。您正在尝试访问 xTest 上的 pred，假设 xTest 是一个数据框，但它只是数据框中的一列。