获取掩码数组的最小值

Question

我有一个掩码数组，我想从中 return 索引的 最小值 。此外，如果有多个，我想 return 随机选择的最小值的索引。在下面的示例中，这应该随机 return 索引 4 或 5:

import numpy as np
import numpy.ma as ma
import random

my_mask = [1, 0, 0, 1, 0, 0]
my_array = [ 0.018, 0.011, 0.004, 0.003, 0.0, 0.0]
masked_array = ma.masked_array(my_array,my_mask)

min_indices = np.where(masked_array.min() == masked_array)
min_index = np.random.choice(min_indices[0])

print masked_array
print min_index

我的问题：被屏蔽的元素被视为零（？），{0,3,4,5}中的任何元素都可以被return编辑。

我的问题：return 数组（不包括屏蔽值）中（随机选择的）最小值的索引的好方法是什么？

Answer 1

使用ma.where()代替np.where()

min_indices = ma.where(masked_array == masked_array.min()))
print(min_indices)

给出：

(array([4, 5]),)

ma 模块有很多函数是为处理掩码数组而设计的。

最后，从这个结果中获取一个随机元素类似于：

min_index = np.random.choice(min_indices[0])

获取掩码数组的最小值

Get minimum value of masked array

python

arrays

numpy

indexing

mask