csv文件划分训练集和测试集

要将CSV文件划分为训练集和测试集，可以按照以下步骤进行操作：

1. 导入CSV文件并将其转换为pandas DataFrame。

import pandas as pd

data = pd.read_csv('data.csv')

2. 将数据集随机划分为训练集和测试集。

from sklearn.model_selection import train_test_split

train_data, test_data = train_test_split(data, test_size=0.2, random_state=42)

其中，test_size表示测试集所占比例，random_state为随机种子，确保每次划分结果一致。

3. 将划分后的数据集保存为CSV文件。

train_data.to_csv('train_data.csv', index=False)
test_data.to_csv('test_data.csv', index=False)

其中，index=False表示不保存索引列。

此时，就可以得到划分好的训练集和测试集CSV文件。