Quickest way to find the nth largest value in a numpy Matrix

前端 未结 6 1870
孤独总比滥情好
孤独总比滥情好 2020-12-29 06:56

There are lots of solutions to do this for a single array, but what about a matrix, such as:

>>> k
array([[ 35,  48,  63],
       [ 60,  77,  96],
          


        
6条回答
  •  有刺的猬
    2020-12-29 07:46

    nums = [[ 35,  48,  63],
            [ 60,  77,  96],
            [ 91, 112, 135]]
    
    highs = [max(lst) for lst in nums]
    highs[nth]
    

提交回复
热议问题