在python中,shape函數的作用是用來返回一個數組或矩陣的維度。對于一維數組,shape函數返回一個整數,表示數組的長度。對于多維數組或矩陣,shape函數返回一個元組,表示各個維度的長度。
例如,對于一個一維數組arr
,可以通過arr.shape
獲取數組的長度。對于一個二維數組或矩陣arr
,可以通過arr.shape
獲取數組的行數和列數。
下面是一些示例:
import numpy as np
arr1 = np.array([1, 2, 3, 4, 5])
print(arr1.shape) # 輸出 (5,)
arr2 = np.array([[1, 2, 3],
[4, 5, 6]])
print(arr2.shape) # 輸出 (2, 3)
arr3 = np.array([[[1, 2, 3],
[4, 5, 6]],
[[7, 8, 9],
[10, 11, 12]]])
print(arr3.shape) # 輸出 (2, 2, 3)
可以看到,shape函數返回的結果是一個元組,元組中的每個元素表示對應維度的長度。對于一維數組,元組只有一個元素,表示數組的長度;對于多維數組或矩陣,元組的長度表示維度的個數,每個元素表示對應維度的長度。