温馨提示×

python .size函数在多维数组中的表现

小樊
85
2024-09-16 13:09:27
栏目: 编程语言

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。

0