для эффективного извлечения патчей, вот так -

я естьnumpy vectorиnumpy array.

Мне нужно взять из каждой строки в матрице первые N (скажем, 3) значений, которые меньше (или равны) соответствующей линии в векторе.

так что, если это мой вектор:

7,
9,
22,
38,
6,
15

и это моя матрица:

[[ 20.,   9.,   7.,   5.,   None,   None],
 [ 33.,  21.,  18.,   9.,   8.,   7.],
 [ 31.,  21.,  13.,  12.,   4.,   0.],
 [ 36.,  18.,  11.,   7.,   7.,   2.],
 [ 20.,  14.,  10.,   6.,   6.,   3.],
 [ 14.,  14.,  13.,  11.,   5.,   5.]]

вывод должен быть:

[[7,5,None],
 [9,8,7],
 [21,13,12],
 [36,18,11],
 [6,6,3],
 14,14,13]]

Есть ли эффективный способ сделать это с помощью масок или чего-то, без отвратительногоfor цикл?

Любая помощь будет оценена!

Ответы на вопрос(1)

Ваш ответ на вопрос