Python 从列表中获取子列表

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

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]

有什么方法可以实现这一点吗？

添加额外的括号：

X[[x for x in range(n_folds) if x != n]]

---

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

在1.6k+rep中，您应该知道您应该共享您得到的错误。