для эффективного извлечения патчей, вот так -
я есть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
цикл?
Любая помощь будет оценена!