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

发布时间：2020-12-16 23:01:18 所属栏目：Python 来源：网络整理

导读：我有一个像这样的数组作为标签列(2个标签：0和1),例如： [0,1,1] 假设我想将此数组转换为具有形状(5,2)(5个元素,2个标签)的numpy矩阵.我怎么能通过使用任何util库以一种微不足道的方式做到这一点？我想要的结果是这样的： [[0,1][1,0],[0,1],[1,0]] 解决方

我有一个像这样的数组作为标签列(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]]

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!