温馨提示×

python中shape函数的作用是什么

小亿
138
2023-12-23 21:38:16
栏目: 编程语言

在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函数返回的结果是一个元组,元组中的每个元素表示对应维度的长度。对于一维数组,元组只有一个元素,表示数组的长度;对于多维数组或矩阵,元组的长度表示维度的个数,每个元素表示对应维度的长度。

0