可以使用numpy.transpose()
函数来调整数组的轴的顺序。该函数接受一个整数元组作为参数,表示要调整的轴的顺序。
例如,将一个2维数组的轴顺序从(0,1)调整为(1,0)可以这样做:
import numpy as np arr = np.array([[1, 2, 3], [4, 5, 6]]) transposed_arr = np.transpose(arr, (1, 0)) print(transposed_arr)
输出结果为:
[[1 4] [2 5] [3 6]]
在这个例子中,原始数组arr
的shape为(2, 3),调整轴的顺序后,新数组transposed_arr
的shape变为(3, 2)。