在NumPy中,我们可以通过以下方式对数组进行排序:
>>> import numpy as np
>>> a = np.array([0, 100, 200])
>>> order = np.array([1, 2, 0])
>>> print(a[order])
[100 200 0]
然而,当“order”是一个矩阵时,这种方法就行不通了:
>>> A = np.array([ [0, 1, 2],
[3, 4, 5],
[6, 7, 8]])
>>> Ord = np.array([ [1, 0, 2],
[0, 2, 1],
[2, 1, 0]])
>>> print(A[Ord].shape)
(3, 3, 3)
我希望将"A"按照以下方式排序:
array([[1, 0, 2],
[3, 5, 4],
[8, 7, 6]])