有效地拆分numpy数组

Question

假设存在对应于训练样本X

的标签向量y

>>> y
array([0, 0, 1, 1, 2, 2])

如果我直接使用numpy.array_split，那就变成

>>> np.array_split(y,2)
[array([0, 0, 1]), array([1, 2, 2])]

如何将y和X分成2个文件夹。所以y成为

>>> y1
array([0, 1, 2])
>>> y2
    array([0, 1, 2])

PS：每个标签具有相同数量的样本。我想将它们等效地分成文件夹。每个文件夹每个标签只有一个样本。

Answer 1

以下解决方案比切片解决方案有效 5倍。所以，我在这里添加它只是为了完整。

GridViewColumn