用python将一个数据文件分为训练集和测试集