我有如下的numpy数组,我想计算columna> 1和columnb <1
的总次数数组
[[1.2, 0.693],
[1.2, 0.724],
[0.976, 0.039],
[0.987, 0.725],
[0.979, 0.528],
[0.978, 0.600],
[0.983, 0.654],
[0.986, 0.059],
[0.979, 0.644],
[0.982, 0.468]])]
答案应为2
答案 0 :(得分:4)
我们可以使用arr[:,0]
获得第一列,而使用arr[:,1]
获得第二列。然后,我们可以使用arr[:,0] > 1
和arr[:,1] < 1
进行检查。通过使用AND运算符(&
),当且仅当两个对应元素为True
时,才能获得其中元素为True
的数组。最后,我们可以对布尔值求和以得出它是True
的次数:
>>> ((arr[:,0] > 1) & (arr[:,1] < 1)).sum()
2