NumPy如何交换数组轴

1292
2024/5/31 12:48:12
栏目: 编程语言
开发者测试专用服务器限时活动,0元免费领,库存有限,领完即止! 点击查看>>

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]]

辰迅云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>

推荐阅读: numpy二维数组切片的原理是什么