要使用 NumPy 庫增強 Python 中的 .size
函數功能,首先需要導入 NumPy 庫
import numpy as np
# 創建一個 NumPy 數組
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 計算數組的元素總數(使用 NumPy 的 size 函數)
total_elements = np.size(arr)
print("Total elements:", total_elements)
# 計算數組的維度
dimensions = arr.ndim
print("Dimensions:", dimensions)
# 計算數組的形狀
shape = arr.shape
print("Shape:", shape)
# 計算數組中每個軸的元素數量
axis_lengths = [arr.shape[i] for i in range(arr.ndim)]
print("Axis lengths:", axis_lengths)
這將輸出:
Total elements: 9
Dimensions: 2
Shape: (3, 3)
Axis lengths: [3, 3]
在這個例子中,我們使用了 NumPy 的 size
函數來計算數組的元素總數。同時,我們還可以使用數組的 ndim
屬性來獲取數組的維度,使用 shape
屬性來獲取數組的形狀,以及通過遍歷形狀元組來獲取每個軸的元素數量。這些信息有助于更好地了解和操作 NumPy 數組。