python - 如何从特定列的值最大的 numpy 数组中选择行？

Question

给定一个特定形状的数组 (m, n)，我想选择特定列值最大的整行(或找到其索引)。

考虑下面的数组。我想找到第二列的值最大的行。第二列的最大值是 0.795，所以我应该返回 [0.21212121, 0.795]。

array([[-3.        ,  0.5       ],
       [-2.93939394,  0.5       ],
       [-2.87878788,  0.5       ],
       [ 0.21212121,  0.795     ],
       [ 0.27272727,  0.785     ],
       [ 0.33333333,  0.785     ],
       [ 0.39393939,  0.77      ],
       [ 2.93939394,  0.5       ],
       [ 3.        ,  0.5       ]])

我通过以下方式达到了预期的效果:

best_result = np.max(acc_m[:, 1])
row_with_best_res = acc_m[acc_m[:, 1] == best_result]

其中 acc_m 是数组的名称。

所提供的解决方案有效，但我不敢相信没有更奇特的 pythonic 方法来做到这一点。

Answer 1

使用 np.argmax()

你的情况

row_with_best_res = acc_m[acc_m[:, 1].argmax()]