Определение порога матрицы по пороговому вектору

Question

У меня есть numpy 2D-массив формы [N, M] и 1D-массив пороговых значений формы M. Я хочу установить минус бесконечное значение для каждого элемента в 2D-матрице, если значение в индексе (X , i) (где X может быть любой строкой, X находится между 0 и M-1) ниже, чем значение в векторе порога, в индексе i.

A non-l oop или list- понимание решения будет оценено.

Answer 1

вот фрагмент, который вы можете использовать ...

data=np.random.rand(5,3)
threshold=np.random.rand(3)

data[data-threshold<0]=-np.Inf

Я использовал случайные данные для тестирования, вам не нужны первые две строки, если ваши данные уже существуют.