NumPy可以通过reshape()
方法来改变数组的形状。这个方法会返回一个新的具有指定形状的数组,但是原始数组的数据不会被复制,而是通过改变数组的步长来实现形状的改变。
import numpy as np arr = np.array([[1, 2, 3], [4, 5, 6]]) new_arr = arr.reshape((3, 2)) print(new_arr)
这将输出:
[[1 2] [3 4] [5 6]]
在reshape()
方法中,传入一个元组作为参数,元组的每个元素表示新数组的每个维度的大小。需要注意的是,新数组的元素个数必须和原数组的元素个数一致。