交叉验证数据拆分_牛客题霸

编写一个 Python 函数，从头开始执行 k -Fold 交叉验证数据拆分。该函数输入一个数据集（一个 2维 NumPy 数组，其中每行代表一个数据样本，每列代表一个特征）和代表折叠数量的整数 k。该函数应将数据集分成 k 个部分，一个部分作为测试集，其余部分作为训练集，并返回一个列表，其中每个元素都是一个元组，其中包含每次折叠的训练集和测试集。

输入描述:

第1行输入一个数据集，第2行输入整数k。

输出描述:

输出一个列表，每个元素都是一个元组，其中包含每次折叠的训练集和测试集。

示例1

输入

[[1, 2, 3], [4, 5, 6], [7, 8, 9], [10, 11, 12], [13, 14, 15]]
3

输出

[[[[7, 8, 9], [1, 2, 3], [10, 11, 12]], [[4, 5, 6], [13, 14, 15]]], [[[4, 5, 6], [13, 14, 15], [10, 11, 12]], [[7, 8, 9], [1, 2, 3]]], [[[4, 5, 6], [13, 14, 15], [7, 8, 9], [1, 2, 3]], [[10, 11, 12]]]]

备注:

1.对应的输入、输出已给出，您只用实现核心功能函数即可。2.支持numpy、scipy、pandas、scikit-learn库。