Làm thế nào để chuyển đổi một lớp cột nhị phân để NumPy mảng

Question

Tôi có một mảng như thế này như cột nhãn (2 nhãn: 0 và 1), ví dụ:

[0,1,0,1,1] 

Giả sử tôi muốn chuyển đổi mảng này đến một ma trận sumpy với hình dạng (5,2) (5 yếu tố, 2 nhãn). Làm thế nào tôi có thể làm điều đó một cách tầm thường bằng cách sử dụng bất kỳ thư viện util?

Kết quả tôi muốn là như thế này:

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

Answer 1

Bạn có thể sử dụng NumPy broadcasting -

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

mẫu chạy -

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