Python数组shape

2023-11-19 11:29:17

在Python中，数组的形状(shape)指的是数组的维度和大小。可以使用numpy库来处理数组的形状。

下面是一些常见的处理数组形状的方法:

1. 使用shape属性:可以使用数组的shape属性来获取数组的形状。例如，对于一个二维数组arr，可以使用arr.shape来获取其形状。

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
print(arr.shape)  # 输出 (2, 3)

2. 使用reshape方法:可以使用reshape方法来改变数组的形状。reshape方法接受一个元组作为参数，指定新的形状。注意，新的形状的元素个数必须与原数组的元素个数相同。

import numpy as np

arr = np.array([1, 2, 3, 4, 5, 6])
new_arr = arr.reshape((2, 3))
print(new_arr)  # 输出 [[1 2 3]
                #      [4 5 6]]

3. 使用resize方法:resize方法与reshape方法类似，也可以改变数组的形状。不同之处在于，resize方法会直接修改原数组的形状。

import numpy as np

arr = np.array([1, 2, 3, 4, 5, 6])
arr.resize((2, 3))
print(arr)  # 输出 [[1 2 3]
            #      [4 5 6]]

4. 使用flatten方法:flatten方法可以将多维数组转换为一维数组。

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
new_arr = arr.flatten()
print(new_arr)  # 输出 [1 2 3 4 5 6]

这些方法可以帮助你处理数组的形状，使其满足你的需求。