python - 在numpy数组中查找最大元素的索引，不包括少数索引

Question

说:

p = array([4, 0, 8, 2, 7])

想要找到最大值的索引，除了少数索引，说:

excptIndx = [2, 3]

Ans: 4，因为 7 将是最大的。

如果 excptIndx = [1, 3] , Ans: 2, 因为 8 是最大的。

Answer 1

在 numpy 中，您可以屏蔽 excptIndx 处的所有值并运行 argmax获取最大元素的索引:

import numpy as np

p = np.array([4, 0, 8, 2, 7])
excptIndx = [2, 3]

m = np.zeros(p.size, dtype=bool)
m[excptIndx] = True
a = np.ma.array(p, mask=m)
print(np.argmax(a))
# 4