在Python中,.size
函數通常用于獲取NumPy數組或Pandas DataFrame等數據結構的大小
import numpy as np
custom_size()
,它接受一個NumPy數組作為輸入參數:def custom_size(array):
# 計算數組的形狀(行數和列數)
shape = array.shape
# 計算數組的大小
size = np.prod(shape)
return size
# 創建一個示例數組
example_array = np.array([[1, 2, 3], [4, 5, 6]])
# 調用自定義函數并打印結果
print(custom_size(example_array))
這將輸出:
6
請注意,這個示例僅適用于NumPy數組。要為其他數據結構(如Pandas DataFrame)編寫自定義.size
函數,您需要根據該數據結構的特性進行相應的修改。