温馨提示×

NumPy如何交换数组轴

小亿
86
2024-05-13 12:48:12
栏目: 编程语言

NumPy提供了np.swapaxes()函数来交换数组的轴。该函数接受两个参数,分别表示要交换的两个轴的索引。例如,对于一个二维数组,可以使用np.swapaxes(arr, 0, 1)来交换其行和列。

示例代码:

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
print("原始数组:")
print(arr)

arr_swapped = np.swapaxes(arr, 0, 1)
print("交换轴后的数组:")
print(arr_swapped)

输出结果:

原始数组:
[[1 2 3]
 [4 5 6]]
交换轴后的数组:
[[1 4]
 [2 5]
 [3 6]]

0