在python中,shape函數(shù)的作用是用來返回一個數(shù)組或矩陣的維度。對于一維數(shù)組,shape函數(shù)返回一個整數(shù),表示數(shù)組的長度。對于多維數(shù)組或矩陣,shape函數(shù)返回一個元組,表示各個維度的長度。
例如,對于一個一維數(shù)組arr
,可以通過arr.shape
獲取數(shù)組的長度。對于一個二維數(shù)組或矩陣arr
,可以通過arr.shape
獲取數(shù)組的行數(shù)和列數(shù)。
下面是一些示例:
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函數(shù)返回的結(jié)果是一個元組,元組中的每個元素表示對應(yīng)維度的長度。對于一維數(shù)組,元組只有一個元素,表示數(shù)組的長度;對于多維數(shù)組或矩陣,元組的長度表示維度的個數(shù),每個元素表示對應(yīng)維度的長度。