Python 中的 .size
函數(shù)用于獲取 NumPy 數(shù)組(ndarray)的元素個(gè)數(shù)
以下是一個(gè)使用 .size
函數(shù)的示例:
import numpy as np
# 創(chuàng)建一個(gè) 3x3 的二維數(shù)組
arr_2d = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 獲取二維數(shù)組的元素個(gè)數(shù)
size_2d = arr_2d.size
print("二維數(shù)組的元素個(gè)數(shù):", size_2d) # 輸出:二維數(shù)組的元素個(gè)數(shù): 9
# 創(chuàng)建一個(gè) 2x3x4 的三維數(shù)組
arr_3d = np.random.randint(0, 10, (2, 3, 4))
# 獲取三維數(shù)組的元素個(gè)數(shù)
size_3d = arr_3d.size
print("三維數(shù)組的元素個(gè)數(shù):", size_3d) # 輸出:三維數(shù)組的元素個(gè)數(shù): 24
在這個(gè)示例中,我們首先導(dǎo)入了 NumPy 庫并創(chuàng)建了一個(gè) 3x3 的二維數(shù)組 arr_2d
。然后,我們使用 .size
函數(shù)獲取二維數(shù)組的元素個(gè)數(shù),結(jié)果為 9。接下來,我們創(chuàng)建了一個(gè) 2x3x4 的三維數(shù)組 arr_3d
,并使用 .size
函數(shù)獲取其元素個(gè)數(shù),結(jié)果為 24。