我对 numpy 和科学计算相当陌生,我为一个问题苦苦挣扎了好几天,所以我决定将其发布在这里。
我正在尝试获取 numpy 数组中某个条件的特定出现次数的计数。
In [233]: import numpy as np
In [234]: a= np.random.random([5,5])
In [235]: a >.7
Out[235]: array([[False, True, True, False, False],
[ True, False, False, False, True],
[ True, False, True, True, False],
[False, False, False, False, False],
[False, False, True, False, False]], dtype=bool)
Run Code Online (Sandbox Code Playgroud)
我想计算True每行中出现的次数并在该计数达到某个阈值时保留行:
前任 :
results=[]
threshold = 2
for i,row in enumerate(a>.7):
if len([value for value in row if value==True]) > threshold:
results.append(i) # keep ids for each row that …Run Code Online (Sandbox Code Playgroud)