问Numpy:如何从每一行检索k max值？
EN

Stack Overflow用户

提问于 2010-08-17 05:42:57

回答 1查看 621关注 0票数 1

我想从numpy数组中的每一行中检索k个最大值。我一直在挖掘文档，但没有找到答案(我可能找错了地方)。有没有人有一个简单的代码片段可以做到这一点？

非常感谢，

迪德里克

回答已采纳

发布于 2010-08-17 05:51:31

这里有一个简单的解决方案：

some_array = numpy.random.randint(0, 10, 10)
top_items = numpy.sort(some_array)[-k:]

或者作为lambda：

max_values = lambda k, array: numpy.sort(array)[-k:]

票数 1

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

原文链接：

https://stackoverflow.com/questions/3497622

复制

相似问题

问Numpy:如何从每一行检索k max值？EN