NumPy的shape()函数是NumPy中的一个重要函数,它可以用来获取数组的形状信息,或者用来改变数组的形状。
使用方法
shape()函数有两种用法:
- 获取数组形状:
# 创建一个数组 a = np.array([[1, 2, 3], [4, 5, 6]]) # 获取数组的形状 shape = a.shape # 输出结果 print(shape) # (2, 3)
上面的代码中,我们创建了一个二维数组,使用shape()函数获取它的形状,输出结果是(2, 3),表示这个数组有2行3列。
- 改变数组形状:
# 创建一个数组 a = np.array([[1, 2, 3], [4, 5, 6]]) # 改变数组的形状 a.shape = (3, 2) # 输出结果 print(a) # [[1 2] # [3 4] # [5 6]]
上面的代码中,我们创建了一个二维数组,使用shape()函数将它的形状改变为3行2列,输出结果就是改变后的数组。
NumPy的shape()函数可以用来获取数组的形状信息,也可以用来改变数组的形状,使用起来非常方便。