从列表中获取子列表

Question

我正在尝试编写交叉验证代码，需要迭代一组并拆分训练和测试数据。我试图以下面的方式拆分它，但代码不编译

import numpy as np
X = np.array([[1,2], [3,4], [5,6]])
n_folds = 3

for n in range(n_folds):
    test_fold = X[n]
    train_folds = X[x for x in range(n_folds) if x != n]
    print train_folds, test_fold

预期输出

[3,4,5,6], [1,2]
[1,2,5,6], [3,4]
[1,2,3,4], [5,6]

有没有办法实现这个？

Answer 1

添加额外的括号：

mousemove

内部括号创建一个列表（理解），外部括号告诉numpy使用该列表进行高级索引。