在Python中,shape函數常用于獲取數組的維度信息。在numpy庫中,可以使用shape函數來獲取數組的形狀。下面是shape函數的用法示例:
import numpy as np
# 創建一個二維數組
arr = np.array([[1, 2, 3], [4, 5, 6]])
print(arr.shape) # 輸出 (2, 3),表示數組的形狀是2行3列
# 創建一個三維數組
arr = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])
print(arr.shape) # 輸出 (2, 2, 2),表示數組的形狀是2個2行2列的子數組
# 創建一個一維數組
arr = np.array([1, 2, 3, 4, 5])
print(arr.shape) # 輸出 (5,),表示數組的形狀是一個包含5個元素的一維數組
注意,shape函數返回的結果是一個元組,其中的每個元素表示數組在對應維度上的大小。對于二維數組,(2, 3)表示2行3列;對于三維數組,(2, 2, 2)表示2個2行2列的子數組;對于一維數組,(5,)表示一個包含5個元素的一維數組。