Ben etiket sütunu olarak bu gibi bir dizi (2 etiket: 0 ve 1) sahip

Question

numpy diziye bir ikili sınıflar sütunu dönüştürmek için: Bu diziyi dönüştürmek istediğiniz gerekiyordu

[0,1,0,1,1] 

, örneğin şekil (5,2) ile bir numpy matrisine (5 eleman, 2 etiket). Herhangi bir kullanım kitaplığını kullanarak bunu nasıl daha basit bir şekilde yapabilirim?

istediğim sonuç şu şekildedir:

[[0,1][1,0],[0,1],[1,0],[1,0]] 

Answer 1

Şunları kullanabilirsiniz NumPy broadcasting -

(a[:,None] != np.arange(2)).astype(int) 

Numune koşmak -

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

In [8]: (a[:,None] != np.arange(2)).astype(int) 
Out[8]: 
array([[0, 1], 
     [1, 0], 
     [0, 1], 
     [1, 0], 
     [1, 0]]) 

# Convert to list if needed 
In [14]: (a[:,None] != np.arange(2)).astype(int).tolist() 
Out[14]: [[0, 1], [1, 0], [0, 1], [1, 0], [1, 0]]