将NumPy数组转换为列向量

cs7cruho  于 2023-05-17  发布在  其他
关注(0)|答案(1)|浏览(146)

将DataFrame的列作为子集,以NumPy数组的形式给出y(依赖)变量。
y = train['Survived']
但是打印变量y(y.shape).shape输出(891,)(注意它不是(891, 1),一个列向量)。
我想使用np.matmul执行y与大小为(1 x 10)的变量的矩阵乘法,但它向我抛出了这个错误:
Exception: Dot product shape mismatch, (891,) vs (1, 10)
如何将y变量强制为大小为(891, 1)的列向量,而不仅仅是(891, )

6tr1vspr

6tr1vspr1#

使用y[:,None]。这将有正确的形状

相关问题