NumPy如何調(diào)整數(shù)組軸的順序

小億
117
2024-05-13 12:50:11

可以使用numpy.transpose()函數(shù)來(lái)調(diào)整數(shù)組的軸的順序。該函數(shù)接受一個(gè)整數(shù)元組作為參數(shù)，表示要調(diào)整的軸的順序。

例如，將一個(gè)2維數(shù)組的軸順序從(0,1)調(diào)整為(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)

輸出結(jié)果為：

[[1 4]
 [2 5]
 [3 6]]

在這個(gè)例子中，原始數(shù)組arr的shape為(2, 3)，調(diào)整軸的順序后，新數(shù)組transposed_arr的shape變?yōu)?3, 2)。

0 踩