我有一个像这样的数组作为标签列(2个标签:0和1),例如:
[0,1,1]
假设我想将此数组转换为具有形状(5,2)(5个元素,2个标签)的numpy矩阵.我怎么能通过使用任何util库以一种微不足道的方式做到这一点?
我想要的结果是这样的:
[[0,1][1,0],[0,1],[1,0]]
解决方法
你可以使用
NumPy broadcasting
–
(a[:,None] != np.arange(2)).astype(int)
样品运行 –
In [7]: a = np.array([0,1]) In [8]: (a[:,None] != np.arange(2)).astype(int) Out[8]: array([[0,0]]) # Convert to list if needed In [14]: (a[:,None] != np.arange(2)).astype(int).tolist() Out[14]: [[0,0]]