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

Stack Overflow用户

提问于 2021-11-01 12:19:15

回答 1查看 55关注 0票数 2

例如，我有：

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

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

我要：

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

非常感谢！！

python

numpy

回答 1

Stack Overflow用户

回答已采纳

发布于 2021-11-01 12:23:32

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

票数 6

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/69796834

复制

相似问题

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

回答 1

Stack Overflow用户

社区

活动

圈层

关于

腾讯云开发者

热门产品

热门推荐

更多推荐

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

回答 1

Stack Overflow用户

社区

活动

圈层

关于

腾讯云开发者

热门产品

热门推荐

更多推荐

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