Question

如何在numpy矩阵中提取哪些元素低于阈值的索引，然后在另一个相同大小的矩阵中将此索引设置为零值？

例如：

input = [2 3 4 1 6 2 1 7 8]
other_same_array = [12 2 -1 4 2 6 9 3 1]

if threshold = 3
output = [0 2 -1 0 2 0 0 3 1]

这是我的代码：

      m =  np.where(input < threshold)
      other_same_array[m] = 0

Answer 1

您甚至可能不需要索引，只需使用布尔掩码来索引输出数组：

mask = input < 3
output[mask] = 0

我希望有所帮助。