Numpy中的形状(4,)和(4,1)之间有什么区别?

时间:2012-11-03 18:31:22

标签: python numpy scipy

我有两个ndarray AB,其中一个的形状为(4,),另一个为(4,1)

当我想使用this计算余弦距离时,会抛出一些抱怨两个对象的异常are not aligned

有没有人有这方面的想法?谢谢!

1 个答案:

答案 0 :(得分:8)

一个是一维数组,另一个是二维数组。

示例:

>>> import numpy as np
>>> a = np.arange(4).reshape(4,1)
>>> a
array([[0],
       [1],
       [2],
       [3]])
>>> a.ravel()
array([0, 1, 2, 3])
>>> a.squeeze()
array([0, 1, 2, 3])
>>> a[:,0]
array([0, 1, 2, 3])
>>>
>>> a[:,0].shape
(4,)