python简单获取数组元素个数的方法

当我们在使用Python编程时，经常会遇到需要获取数组中元素的数量的情况。这里列举了三种获取数组元素数量的方法。

方法1: len()函数

在Python中，可以使用内置函数len()来获取数组/列表的元素个数。

# 示例1：使用len()函数获取列表的元素个数
my_list = [1, 2, 3, 4, 5]
list_length = len(my_list)
print("列表my_list中的元素个数为: ", list_length)

输出结果：

列表my_list中的元素个数为: 5

# 示例2：使用len()函数获取数组的元素个数
import numpy as np
my_array = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
array_length = len(my_array)
print("数组my_array中的元素个数为: ", array_length)

输出结果：

数组my_array中的元素个数为: 3

方法2: np.size()函数

在使用NumPy库时，可以使用numpy.size()函数来获取数组的元素数量。

# 示例3：使用np.size()函数获取数组的元素个数
import numpy as np
my_array = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
array_size = np.size(my_array)
print("数组my_array中的元素个数为: ", array_size)

输出结果：

数组my_array中的元素个数为: 9

方法3: shape属性

在使用NumPy库时，还可以使用数组的shape属性来获取该数组的形状，即每个轴上的元素数量。若该数组是个一维数组，则shape返回一个整数n，表示该数组的长度为n；如果是二维数组，则shape返回一个长度为2的元组(m, n)，其中m表示行数，n表示列数，以此类推。

# 示例4：使用shape属性获取数组的元素个数
import numpy as np
my_array = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
array_shape = my_array.shape
print("数组my_array的形状为: ", array_shape)
array_size = my_array.size
print("数组my_array中的元素个数为: ", array_size)

输出结果：

数组my_array的形状为: (3, 3)
数组my_array中的元素个数为: 9

通过以上三种方法，我们可以轻松地获取数组/列表的元素数量。