NumPy如何调整数组轴的顺序

numpy

851

2024/5/25 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)。

最新知识库