获取Numpy数组的长度

获取Numpy数组的长度

参考：Numpy Array Length

在Numpy中，数组是一种用于存储相同类型的数据的数据结构。数组的长度是指数组中元素的个数，可以通过shape属性或者len()函数来得到数组的长度。

使用shape属性获取数组长度

Numpy中的数组可以是多维的，所以shape属性返回的是一个包含各个维度大小的元组。

获取一维数组长度

若数组为一维数组，则shape属性返回一个只包含一个元素的元组，元素值即为数组的长度。

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
print(arr.shape[0])

运行结果为：

获取多维数组长度

多维数组的长度可以通过取该组的第一个元素来获得。示例代码如下：

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
print(arr.shape[0])

输出结果：

使用len()函数获取数组长度

除了shape属性，也可以使用Python内置的len()函数来获取数组的长度，len()函数返回数组的第一个维度的大小。

获取一维数组的长度

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
print(len(arr))

运行结果为：

获取多维数组的长度

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
print(len(arr))

运行结果为：

总结

无论使用shape属性还是len()函数，都可以很方便地得到Numpy数组的长度。在实际应用中，可以根据需要选择合适的方法来获取数组的长度。