我想从numpy数组中的每一行检索k max值。我一直在挖掘文档，但找不到答案（我可能在错误的地方寻找）。有没有人有一个简单的代码片段可以做到这一点？
非常感谢，
Diederik
答案 0 :(得分：1)
这是一个简单的解决方案：
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:]