将多个numpy列组合成一个用于多标签分类

Question

我有一个包含多个列的数组，其中每列代表一个虚拟变量。我需要将所有列组合成单个载体以准备多标签分类。因此，nx6数组需要变成nx1

的数组

例如：

array([[ 0.,  1.,  0.,  0.,  0.,  1.],
       [ 0.,  0.,  0.,  1.,  0.,  1.],
       [ 0.,  0.,  0.,  1.,  1.,  0.]])

应该是这样的：

array([[ 0.  1.  0.  0.  0.  1.],
       [ 0.  0.  0.  1.  0.  1.],
       [ 0.  0.  0.  1.  1.  0.]])

现在我有六列，但我想要为任意数量的列数组推广代码，因为我不确定我们最终会使用多少个标签。

到目前为止，我已经尝试了numpy.column_stack，但仍然给了我一个nx6形状的数组。有任何想法吗？

谢谢！