python - 为什么一维数组的形状不显示行数为 1？

Question

我知道 numpy 数组有一个名为 shape 的方法，它返回 [No.of rows, No.of columns]，shape[0] 为您提供行数， shape[1] 为您提供列数。

a = numpy.array([[1,2,3,4], [2,3,4,5]])
a.shape
>> [2,4]
a.shape[0]
>> 2
a.shape[1]
>> 4

但是，如果我的数组只有一行，那么它会返回 [No.of columns, ]。并且 shape[1] 将超出索引。例如

a = numpy.array([1,2,3,4])
a.shape
>> [4,]
a.shape[0]
>> 4    //this is the number of column
a.shape[1]
>> Error out of index

如果数组可能只有一行，我如何获取 numpy 数组的行数？

谢谢

Answer 1

rows 和 columns 的概念适用于二维数组。但是，数组 numpy.array([1,2,3,4]) 是一维数组，因此只有一维，因此 shape 正确地返回单值可迭代的。

对于同一数组的 2D 版本，请考虑以下情况:

>>> a = numpy.array([[1,2,3,4]]) # notice the extra square braces
>>> a.shape
(1, 4)