在python中,shape函数的作用是用来返回一个数组或矩阵的维度。对于一维数组,shape函数返回一个整数,表示数组的长度。对于多维数组或矩阵,shape函数返回一个元组,表示各个维度的长度。
例如,对于一个一维数组arr
,可以通过arr.shape
获取数组的长度。对于一个二维数组或矩阵arr
,可以通过arr.shape
获取数组的行数和列数。
下面是一些示例:
import numpy as np
arr1 = np.array([1, 2, 3, 4, 5])
print(arr1.shape) # 输出 (5,)
arr2 = np.array([[1, 2, 3],
[4, 5, 6]])
print(arr2.shape) # 输出 (2, 3)
arr3 = np.array([[[1, 2, 3],
[4, 5, 6]],
[[7, 8, 9],
[10, 11, 12]]])
print(arr3.shape) # 输出 (2, 2, 3)
可以看到,shape函数返回的结果是一个元组,元组中的每个元素表示对应维度的长度。对于一维数组,元组只有一个元素,表示数组的长度;对于多维数组或矩阵,元组的长度表示维度的个数,每个元素表示对应维度的长度。