温馨提示×

NumPy如何调整数组轴的顺序

小亿
121
2024-05-13 12:50:11
栏目: 编程语言

可以使用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)。

0