python中的shape函数

在numpy模块中，shape函数用于获取数组的形状。它返回一个由整数构成的元组，表示数组在各个维度上的大小。比如，对于一个二维数组，shape函数返回一个包含两个整数的元组，第一个整数表示数组的行数，第二个整数表示数组的列数。

示例代码：

import numpy as np

a = np.array([[1, 2, 3], [4, 5, 6]])
print(a.shape)  # 输出 (2, 3)

b = np.array([1, 2, 3, 4])
print(b.shape)  # 输出 (4,)

在上面的代码中，a是一个二维数组，其形状为(2, 3)，即有2行3列。b是一个一维数组，其形状为(4,)，即有4个元素。