NumPy：在矩阵中找到 N 个最大元素

Posted 2023-02-23

【中文标题】NumPy：在矩阵中找到 N 个最大元素

【英文标题】：NumPy: finding N largest elements in a matrix [duplicate]

【发布时间】：2015-06-24 14:45:28

【问题描述】：

编辑，因为我的上一个问题是重复的，但我也在努力解决这个问题。我目前正在使用矩阵，可以轻松找到最大的元素

M[M != 1].max()

但是，我对获取最大的 N 个元素很感兴趣，但找不到使用矩阵的简单方法。有没有有效的解决方案？

【问题讨论】：

嗨@planner15 - 根据链接，np.where(M == M[M != 0].min()) 应该返回对应于最小元素的行和列索引。

修改后的问题仍然是重复的，请参阅 this question 和 this question 查找索引。

@moarningsun 我看到了那些，但那些是用于数组的。它们仍然适用于矩阵吗？

@planner15：同样的基本思想适用，但您可能需要做一些小的语法更改。

【参考方案1】：

是的，有一个where 方法将条件作为参数之一，

minimum = M[M != 0].min()

print numpy.where(M==minimum)