python - 如何将二进制类列转换为 numpy 数组

Question

我有一个像这样的数组作为标签列(2 个标签:0 和 1)，例如:

[0,1,0,1,1]

假设我想将此数组转换为形状为 (5,2) (5 elements, 2 labels) 的 numpy 矩阵。我怎样才能使用任何 util 库以简单的方式做到这一点？

我想要的结果是这样的:

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

Answer 1

你可以使用 NumPy broadcasting -

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

sample 运行-

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