NumPy 陣列形狀
陣列的形狀
陣列的形狀是指每個維度中的元素數量。
獲取陣列的形狀
NumPy 陣列有一個名為 shape
的屬性,它返回一個元組,其中每個索引都包含相應維度的元素數量。
示例
列印一個二維陣列的形狀
import numpy as np
arr = np.array([[1, 2, 3, 4], [5, 6, 7, 8]])
print(arr.shape)
自己動手試一試 »
上面的示例返回 (2, 4)
,這意味著該陣列有 2 個維度,第一個維度有 2 個元素,第二個維度有 4 個元素。
示例
使用 ndmin
建立一個具有 5 個維度的陣列,使用值為 1,2,3,4 的向量,並驗證最後一個維度的值為 4
import numpy as np
arr = np.array([1, 2, 3, 4], ndmin=5)
print(arr)
print('陣列的形狀 :', arr.shape)
自己動手試一試 »
形狀元組代表什麼?
每個索引處的整數告訴我們相應維度有多少個元素。
在上面的例子中,索引 4 處的值為 4,所以我們可以說第 5 個(4+1 個)維度有 4 個元素。