Stellen Teilmenge von Array, basierend auf den Werten von zwei anderen Arrays in Python

Question

ich Python verwende. Wie man eine Unterauswahl eines Vektors macht, basierend auf den Werten von zwei anderen Vektoren mit der gleichen Länge?

Zum Beispiel dieser drei Vektoren

c1 = np.array([1,9,3,5]) 
c2 = np.array([2,2,3,2]) 
c3 = np.array([2,3,2,3]) 

c2==2 
array([ True, True, False, True], dtype=bool) 
c3==3 
array([False, True, False, True], dtype=bool) 

ich so etwas wie dies tun will:

elem = (c2==2 and c3==3) 
c1sel = c1[elem] 

Aber die erste Anweisung führt zu einem Fehler:

Traceback (most recent call last): 
File "<stdin>", line 1, in <module> 
ValueError: The truth value of an array with more than one element is ambiguous. 
Use a.any() or a.all() 

In Matlab, Ich würde verwenden:

elem = find(c2==2 & c3==3); 
c1sel = c1(elem); 

Wie dies in Python zu tun?

Answer 1

Sie numpy.logical_and verwenden können:

>>> c1[np.logical_and(c2==2, c3==3)] 
array([9, 5]) 

Answer 2

Alternativ versuchen

>>> c1[(c2==2) & (c3==3)] 
array([9, 5]) 

cf) Mit Python Operator Precedence ist die Priorität von & oberen als ==. Siehe die folgenden Ergebnisse.

>>> 1 == 1 & 2 == 2 
False 

>>> (1 == 1) & (2 == 2) 
True 

Answer 3

Sie haben alle Ihre Bedingungen in Klammern zu halten:

In []: c1[(c2 == 2) & (c3 == 3)] 
Out[]: array([9, 5])