如何计算第一列> 1和第二列<1的numpy数组中的总时间

时间:2019-11-14 21:46:43

标签: numpy

我有如下的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

1 个答案:

答案 0 :(得分:4)

我们可以使用arr[:,0]获得第一列,而使用arr[:,1]获得第二列。然后,我们可以使用arr[:,0] > 1arr[:,1] < 1进行检查。通过使用AND运算符(&),当且仅当两个对应元素为True时,才能获得其中元素为True的数组。最后,我们可以对布尔值求和以得出它是True的次数:

>>> ((arr[:,0] > 1) & (arr[:,1] < 1)).sum()
2