python - 如何按另一个二维索引数组对二维 numpy 数组的每一行进行排序？

Question

例如，我有:

arr = np.array[[10 30 20],
               [30 20 10]]

indices = np.array[[2 1 0],
                   [2 1 0]]

我要:

[[20 30 10],
 [10 20 30]]

非常感谢!!

Answer 1

使用np.take_along_axis :

import numpy as np

arr = np.array([[10, 30, 20],
               [30, 20, 10]])

indices = np.array([[2, 1, 0],
                   [2, 1, 0]])


res = np.take_along_axis(arr, indices, axis=1)
print(res)

输出

[[20 30 10]
 [10 20 30]]