将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, )
?
1条答案
按热度按时间6tr1vspr1#
使用
y[:,None]
。这将有正确的形状