如何一个二进制类列转换为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]]