python中的shape函数怎么用

在Python中，shape函数常用于获取数组的维度信息。在numpy库中，可以使用shape函数来获取数组的形状。下面是shape函数的用法示例：

import numpy as np
# 创建一个二维数组
arr = np.array([[1, 2, 3], [4, 5, 6]])
print(arr.shape)  # 输出 (2, 3)，表示数组的形状是2行3列
# 创建一个三维数组
arr = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])
print(arr.shape)  # 输出 (2, 2, 2)，表示数组的形状是2个2行2列的子数组
# 创建一个一维数组
arr = np.array([1, 2, 3, 4, 5])
print(arr.shape)  # 输出 (5,)，表示数组的形状是一个包含5个元素的一维数组

注意，shape函数返回的结果是一个元组，其中的每个元素表示数组在对应维度上的大小。对于二维数组，(2, 3)表示2行3列；对于三维数组，(2, 2, 2)表示2个2行2列的子数组；对于一维数组，(5,)表示一个包含5个元素的一维数组。