0 und 1 in einem binären Array invertieren

Question

Gibt es eine Funktion in Numpy, um 0 und 1 in einem binären Array zu invertieren? Wenn

a = np.array([0, 1, 0, 1, 1]) 

möchte ich bekommen:

b = [1, 0, 1, 0, 0] 

Ich benutze:

b[a==0] = 1 
b[a==1] = 0 

aber vielleicht schon es existieren etwas in Numpy dies zu tun.

Answer 1

Sie können einfach tun:

In[1]:b=1-a 
In[2]:b 
Out[2]: array([1, 0, 1, 0, 0]) 

oder

In[22]:b=(~a.astype(bool)).astype(int) 
Out[22]: array([1, 0, 1, 0, 0]) 

Answer 2

Ein funktionaler Ansatz:

>>> np.logical_not(a).astype(int) 
array([1, 0, 1, 0, 0])