用于拆分训练集和测试集的函数 train_test_split