python – 有numpy argsort返回一个2d索引数组吗？

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了python – 有numpy argsort返回一个2d索引数组吗？，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含746字，纯文字阅读大概需要2分钟。

内容图文

如果我们有一个1d数组

arr = np.random.randint(7, size=(5))
# [3 1 4 6 2]
print np.argsort(arr)
# [1 4 0 2 3] <= The indices in the sorted order    

如果我们有一个2d数组

arr = np.random.randint(7, size=(3, 3))
# [[5 2 4]
# [3 3 3]
# [6 1 2]]
print np.argsort(arr)
# [[1 2 0]
# [0 1 2]
# [1 2 0]] <= It sorts each row

我需要的是2d索引,它将整个矩阵排序.像这样的东西：

# [[2 1] => 1
# [0 1] => 2
# [2 2] => 2
# .
# .
# .
# [0 2] => 4
# [0 0] => 5
# [2 0]] => 6

如何获得2d数组排序的“2d索引”？

解决方法:

在flattened数组上应用numpy.argsort,然后将索引解开回(3,3)形状：

>>> arr = np.array([[5, 2, 4],
[3, 3, 3],
[6, 1, 2]])
>>> np.dstack(np.unravel_index(np.argsort(arr.ravel()), (3, 3)))
array([[[2, 1],
        [0, 1],
        [2, 2],
        [1, 0],
        [1, 1],
        [1, 2],
        [0, 2],
        [0, 0],
        [2, 0]]])

内容总结

以上是互联网集市为您收集整理的python – 有numpy argsort返回一个2d索引数组吗？全部内容，希望文章能够帮你解决python – 有numpy argsort返回一个2d索引数组吗？所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。

内容手机端