python - 除了 NumPy 矩阵中的特定列之外，是否有一种有效的方法来获取最大元素的位置？

Question

例如，有一个二维 Numpy 矩阵 M :

[[1,10,3],
 [4,15,6]]

除了 M[:][1] 之外的最大元素是 6 ，其位置为 (1,2) .所以答案是 (1,2) .
非常感谢您的帮助!

Answer 1

单程:

col = 1
skip_col = np.delete(x, col, axis=1)
row, column = np.unravel_index(skip_col.argmax(), skip_col.shape)
if column >= col:
    column += 1 

翻译:

删除列

找到最大参数(argmax 给出扁平结果，unravel_index 给出二维数组中的位置)

如果该列大于或等于跳过的一列，则添加一个

关注 Dunes comment ，我喜欢这个建议。它的行数几乎相同，但不需要副本(如 np.delete)。因此，如果您受内存限制(如在真正的大数据中):

col = 1
row, column = np.unravel_index(x[:, :col].argmax(), x[:, :col].shape)  # left max, saving a line assuming it's the global max, but less readable
right_max = np.unravel_index(x[:, col+1:].argmax(), x[:, col+1:].shape)
if x[right_max] > x[row, column]:
    row, column = right_max
    column += col