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