NumPy如何调整数组轴的顺序
lewis
2024-04-16
36次阅读
可以使用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)。

发表评论:
◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。