NumPy提供了np.swapaxes()函數(shù)來交換數(shù)組的軸。該函數(shù)接受兩個參數(shù)，分別表示要交換的兩個軸的索引。例如，對于一個二維數(shù)組，可以使用np.swapaxes(arr, 0, 1)來交換其行和列。
示例代碼：
import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6]])
print("原始數(shù)組：")
print(arr)
arr_swapped = np.swapaxes(arr, 0, 1)
print("交換軸后的數(shù)組：")
print(arr_swapped)
輸出結(jié)果：
原始數(shù)組：
[[1 2 3]
[4 5 6]]
交換軸后的數(shù)組：
[[1 4]
[2 5]
[3 6]]
億速云公眾號
手機(jī)網(wǎng)站二維碼