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.sizeprint("二维数组的元素个数:", size_2d) # 输出:二维数组的元素个数: 9# 创建一个 2x3x4 的三维数组arr_3d = np.random.randint(0, 10, (2, 3, 4))# 获取三维数组的元素个数size_3d = arr_3d.sizeprint("三维数组的元素个数:", size_3d) # 输出:三维数组的元素个数: 24在这个示例中,我们首先导入了 NumPy 库并创建了一个 3x3 的二维数组 arr_2d。然后,我们使用 .size 函数获取二维数组的元素个数,结果为 9。接下来,我们创建了一个 2x3x4 的三维数组 arr_3d,并使用 .size 函数获取其元素个数,结果为 24。


