python中shape函数的作用是什么

707
2024/1/9 21:38:16
栏目: 编程语言
开发者测试专用服务器限时活动,0元免费领,库存有限,领完即止! 点击查看>>

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

辰迅云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>

推荐阅读: python不同类间参数传递的方法是什么