Как сравнить матричные элементы с порогом в NumPy

Question

У меня есть матрица положительных значений, а также список пороговых значений. Список порогов сортируется. Например:

X = np.matrix([[1., 2., 3.,], [0., 2., 1.], [4., 1., 2.]]) 
y = [0.5, 1., 3., 5.] 

Я хотел бы написать функцию, которая устанавливает все записи X, которые меньше, чем у [1] у [1], то все записи, которые равны или больше, чем у [1] и меньше, чем у [2] на y [2] и т. д.

Это может быть сделано путем простого итерации по элементам матрицы и порогам, конечно, но я хотел бы найти более эффективный способ. Я посмотрел на clip, но это не совсем то, что мне нужно.

Answer 1

Вы можете использовать np.searchsorted для получения индексов:

>>> np.array(y)[np.searchsorted(y, X)] 
array([[ 1. , 3. , 3. ], 
     [ 0.5, 3. , 1. ], 
     [ 5. , 1. , 3. ]])