python

python中的shape函数怎么用

小亿
218
2023-10-10 04:49:52
栏目: 编程语言

在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个元素的一维数组。

0
看了该问题的人还看了