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。