我有两个数组x=[1,2,3,4]
和y=[1,0,0,1]
描述2D点(x,y)。我想知道有多少元素有x>2
和 y==1
。什么是最简单的方法(没有任何循环)?
是否可以执行x[x>2]
之类的操作,但有两种情况?
答案 0 :(得分:2)
假设这些是numpy数组,因为你的x[x>2]
是numpy语法,你只需要和(&
)运算符:
meet_cond = (x > 2) & (y == 1)
how_many = meet_cond.sum()
which_x = x[meet_cond]
which_y = y[meet_cond]