我有一个 numpy向量
和一个 numpy数组
。
我需要从矩阵的每一行中取出前 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
循环?任何帮助都将不胜感激!
None
总是出现在最后。 - Binyamin Even