python - 如何用Python中从某个键开始的另一个数组填充一个NP数组？

Question

我有一个数据框，名为x。它由 2 列组成，如下所示 (712, 2):

     SibSp  Parch
731      0      0
230      1      0
627      0      0
831      1      1
391      0      0
.................

由于逻辑回归需要“自由权重”，我构建了一个 newX 变量，其形状为我的 x 数据框但为空值。

newX = np.zeros(shape=(x.shape[0], x.shape[1] + 1))

这会生成一个 (712, 3) np 数组:

[[0. 0. 0.]
 [0. 0. 0.]
 [0. 0. 0.]
 ...
 [0. 0. 0.]
 [0. 0. 0.]
 [0. 0. 0.]]

由于第一个索引 (0) 是自由权重，我现在想将 x 值分配给索引 1 和 2。

newX[:, 1:] = x

但是，它给了我这个错误:

Exception: Dot product shape mismatch, (712,) vs (3, 712)

如何使用键 1-2 中的 x 数组填充我的 newX NPArray，但保持 0 处的所有键相同？

Answer 1

您可能需要在数据框后添加值

newX[:, 1:] = x.values
newX
Out[171]: 
array([[0., 0., 0.],
       [0., 1., 0.],
       [0., 0., 0.],
       [0., 1., 1.],
       [0., 0., 0.]])