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