python - Numpy:获取数组中的最小值索引，忽略特定值

Question

我正在尝试获取数组中最小值的索引，例如:

ind = np.where((arr == arr.min()))

我想修改它以便忽略特定值。例如:

ind = np.where((arr == arr[arr != value].min()))

上面的解决方案是可以的，但是可以做得更好吗？

Answer 1

您可以创建一个屏蔽数组，然后在其上使用 np.where():

m = np.ma.masked_array(a, mask=(a==value))
np.where(m, m==m.min())

给你想要的。