Python 中的 .size
函数用于获取 NumPy 数组(ndarray)的元素个数
以下是一个使用 .size
函数的示例:
import numpy as np
# 创建一个 3x3 的二维数组
arr_2d = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 获取二维数组的元素个数
size_2d = arr_2d.size
print("二维数组的元素个数:", size_2d) # 输出:二维数组的元素个数: 9
# 创建一个 2x3x4 的三维数组
arr_3d = np.random.randint(0, 10, (2, 3, 4))
# 获取三维数组的元素个数
size_3d = arr_3d.size
print("三维数组的元素个数:", size_3d) # 输出:三维数组的元素个数: 24
在这个示例中,我们首先导入了 NumPy 库并创建了一个 3x3 的二维数组 arr_2d
。然后,我们使用 .size
函数获取二维数组的元素个数,结果为 9。接下来,我们创建了一个 2x3x4 的三维数组 arr_3d
,并使用 .size
函数获取其元素个数,结果为 24。