如何通过另一个 2d 索引数组对 2d numpy-array 的每一行进行排序？

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