如何一個二進制類列轉換爲numpy的陣列

Question

我有這樣的標籤列的陣列（2個標籤：0和1），例如：

[0,1,0,1,1] 

假定欲這個數組轉換到一個形狀爲（5,2）（5個元素，2個標籤）的numpy矩陣。我可以通過使用任何util庫以簡單的方式做到這一點？

我想要的結果是這樣的：

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

Answer 1

你可以使用NumPy broadcasting -

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

採樣運行 -

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]]