Как извлечь индексы того, какие элементы ниже порогового значения в 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
