NumPy 數組形狀

數組的形狀

數組的形狀是每個維中元素的數量。

獲取數組的形狀

NumPy 數組有一個名為 shape 的屬性，該屬性返回一個元組，每個索引具有相應元素的數量。

實例

打印 2-D 數組的形狀：

import numpy as np
arr = np.array([[1, 2, 3, 4], [5, 6, 7, 8]])
print(arr.shape)

運行實例

上面的例子返回 (2, 4)，這意味著該數組有 2 個維，每個維有 4 個元素。

實例

利用 ndmin 使用值 1,2,3,4 的向量創建有 5 個維度的數組，并驗證最后一個維度的值為 4：

import numpy as np
arr = np.array([1, 2, 3, 4], ndmin=5)
print(arr)
print('shape of array :', arr.shape)

運行實例

元組的形狀代表什么？

每個索引處的整數表明相應維度擁有的元素數量。

上例中的索引 4，我們的值為 4，因此可以說第 5 個 ( 4 + 1 th) 維度有 4 個元素。