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

前端之家收集整理的这篇文章主要介绍了如何将二进制类列转换为numpy数组前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。
我有一个像这样的数组作为标签列(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]]
原文链接:https://www.f2er.com/python/241820.html

猜你在找的Python相关文章