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 个元素。