獲取數(shù)組的形狀

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

實例

打印 2-D 數(shù)組的形狀：

import numpy as np

arr = np.array([[1, 2, 3, 4], [5, 6, 7, 8]])

print(arr.shape)

運行實例

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

實例

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

import numpy as np

arr = np.array([1, 2, 3, 4], ndmin=5)

print(arr)
print('shape of array :', arr.shape)

運行實例