学习笔记：使用python将数据集划分成测试集和训练集