2D空間のPythonフィルタポイント

Question

2D点（x、y）を記述する2つの配列x=[1,2,3,4]とy=[1,0,0,1]があります。いくつの要素があるか知りたいですx>2とy==1です。これを行う最も簡単な方法は何ですか（ループなし）。 x[x>2]のようにすることはできますか？

Answer 1

あなたx[x>2]がnumpyの構文であることから、これらは、numpyの配列であると仮定すると、あなただけのと（&）演算子必要があります：xとyがポイントとして一緒に属している場合は、あなたがにそれらをパックしたい場合があります

meet_cond = (x > 2) & (y == 1) 
how_many = meet_cond.sum() 

which_x = x[meet_cond] 
which_y = y[meet_cond] 

Answer 2

をa np 2D配列：

>>> import numpy as np 
>>> x = np.array([1, 2, 3, 4]) 
>>> y = np.array([1, 0, 0, 1]) 
>>> xy = np.array([x, y]).T 
>>> xy[(x > 2) & (y == 1)] 
array([[4, 1]]) 
>>> xy[(xy[:, 0] > 2) & (xy[:, 1] == 1)] 
array([[4, 1]]) 
>>> np.count_nonzero((xy[:, 0] > 2) & (xy[:, 1] == 1)) 
1