获取Numpy Array的长度
导言
Numpy
是一个强大的 Python 库,用于科学计算和数据分析。在 Numpy
中,我们经常需要知道数组(array
)的长度,也就是元素的个数。本文将详细介绍如何获取 Numpy
数组的长度。
数组长度的概念
数组的长度是指数组中元素的个数。在 Numpy
中,数组可以是一维的,也可以是多维的。对于一维数组,长度就是数组中元素的个数。对于多维数组,长度等于数组所有维度的元素个数的乘积。
获取一维数组长度
要获取一维数组的长度,可以使用 len()
函数。以下是一个示例代码:
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
length = len(arr)
print("Array:", arr)
print("Length:", length)
代码运行结果:
在上面的示例中,我们创建了一个包含 5 个元素的一维数组 arr
。然后使用 len()
函数获取了数组的长度,并将结果存储在变量 length
中。最后,打印了数组和长度。
获取多维数组长度
对于多维数组,可以使用 shape
属性来获取数组的维度信息。shape
属性返回一个元组,其中包含各个维度的大小。要获取数组的长度,只需将元组中的各个维度大小相乘。以下是一个示例代码:
import numpy as np
arr = np.array([[1, 2, 3, 4], [5, 6, 7, 8]])
length = np.prod(arr.shape)
print("Array:", arr)
print("Length:", length)
代码运行结果:
在上面的示例中,我们创建了一个包含 2 行 4 列的二维数组 arr
。然后使用 shape
属性获取了数组的维度信息,再使用 np.prod()
函数计算了数组的长度,并将结果存储在变量 length
中。最后,打印了数组和长度。
总结
本文介绍了如何获取 Numpy
数组的长度。对于一维数组,可以使用 len()
函数获取长度;对于多维数组,可以使用 shape
属性获取维度信息,并将各个维度的大小相乘来计算长度。