将multiply numpy列合并为一列进行多标签分类

Question

我有一个由多个列组成的数组，其中每一列表示一个伪变量。我需要将所有列组合成一个向量，为多标签分类做准备。所以nx6的数组需要变成nx1的数组

例如：

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

应该看起来像：

^{pr2}$

但我现在不想用6列的数组来概括任何列的数目。在

到目前为止我已经试过了numpy.column_堆栈但这还是给了我一个nx6形状的数组。有什么想法吗？在

谢谢！在