numpy中的shape属性

.shape 不是一个函数，而是numpy的一个属性（attribute），用于获取数组维度信息。它返回一个元组（tuple），元组中的每个元素代表对应维度的大小。

import numpy as np

# 1D 数组 (向量)
arr1d = np.array([1, 2, 3, 4, 5])
print(f"数组内容: {arr1d}")
print(f"数组的形状: {arr1d.shape}")
# 输出: (5,)
# 解释: 这是一个一维数组，有 5 个元素。

# 2D 数组 (矩阵)
arr2d = np.array([[1, 2, 3],
                  [4, 5, 6]])
print(f"\n数组内容:\n{arr2d}")
print(f"数组的形状: {arr2d.shape}")
# 输出: (2, 3)
# 解释: 这是一个二维数组，有 2 行 3 列。

# 3D 数组
arr3d = np.array([[[1, 2], [3, 4]],
                  [[5, 6], [7, 8]]])
print(f"\n数组内容:\n{arr3d}")
print(f"数组的形状: {arr3d.shape}")
# 输出: (2, 2, 2)
# 解释: 这是一个三维数组，可以理解为 2 个 2x2 的矩阵。