如何在Python中将数据分成10折并保存到数组中

from sklearn.model_selection import KFold

# 假设有一个数据集 data，包含了所有的数据
data = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

# 定义一个空数组，用于保存分割后的数据
folds = []

# 使用 KFold 进行数据分割，n_splits=10 表示将数据分成10折
kf = KFold(n_splits=10)

# 遍历每一折的训练集和测试集
for train_index, test_index in kf.split(data):
    train_data = [data[i] for i in train_index]
    test_data = [data[i] for i in test_index]
    fold = {'train': train_data, 'test': test_data}
    folds.append(fold)

# 打印每一折的训练集和测试集
for i, fold in enumerate(folds):
    print(f"Fold {i+1}:")
    print("Train data:", fold['train'])
    print("Test data:", fold['test'])
    print()