如何获取多维numpy数组

时间:2015-04-22 14:15:57

标签: python arrays algorithm numpy multidimensional-array

我想获得多维numpy数组中最大n值的索引。为了得到一维numpy数组中最大n个值的索引,我找到了this。在python中的交互式shell中测试之后,bottleneck.argpartsort似乎不会对多维numpy数组产生影响。为了获得多维numpy数组中最大值的索引,我找到了this。它无法获得最大的n。我可以给出的方法是将多维numpy数组转换为{value:index}列表(元组存在的索引),然后按值对列表进行排序,并获取它的索引。有没有更容易或更多的表现?

1 个答案:

答案 0 :(得分:4)

我无法访问.htaccess,因此在此示例中我使用bottleneck,但您应该能够以相同的方式使用它:

argsort