在Python中,我們經(jīng)常會用到數(shù)組來存儲各種數(shù)據(jù)。但是,有時候我們需要知道一個數(shù)組的維度,以便于更好地處理它。
那么,如何用Python求一個數(shù)組的維度呢?其實,Python提供了一個很方便的工具——numpy庫。下面就是使用numpy庫中的shape函數(shù)來求一個數(shù)組的維度的例子:
import numpy as np # 定義一個三維數(shù)組 arr = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]], [[9, 10], [11, 12]]]) # 求數(shù)組的維度 dim = arr.shape # 輸出數(shù)組的維度 print(dim)
上面的代碼中,我們定義了一個三維數(shù)組arr,并使用shape函數(shù)求出了數(shù)組的維度。其結(jié)果為(3, 2, 2),也就是說,這個數(shù)組是一個3×2×2的三維數(shù)組。
需要注意的是,如果我們將一個一維數(shù)組作為參數(shù)傳入shape函數(shù),那么得到的結(jié)果將只有一個數(shù)字,表示該數(shù)組的長度。例如:
# 定義一個一維數(shù)組 arr = np.array([1, 2, 3, 4]) # 求數(shù)組的維度 dim = arr.shape # 輸出數(shù)組的維度 print(dim)
輸出結(jié)果為(4,),說明該數(shù)組是一個長度為4的一維數(shù)組。
總之,在Python中求一個數(shù)組的維度非常簡單,只需使用numpy庫中的shape函數(shù)。