如何添加熊猫?作为test_data创建的文件的数据框架?

Question

我将机器学习的训练数据和测试数据分割成这样

train_df, test_df = train_test_split(df, test_size=0.2, random_state=42)

，现在我想保存每个数据到'train_data'和'test_data'文件这些都是可以接近的。/train_data'或'./test_data'

但我不知道怎么做。我发现有'to_csv'，但它不是为这个我想，因为

test_df.to_csv('./test_data')

我得到错误说IsADirectoryError: [Errno 21]是一个目录:'./test_data'。我该怎么办?

Answer 1

import pandas as pd
from sklearn.model_selection import train_test_split
#creating your dataframe
df = ...
train_data, test_data = train_test_split(df, test_size=0.2, random_state=42)
train_data.to_csv(open('train_data.csv','w'))
test_data.to_csv(open('test_data.csv','w'))
#you can optionally add an encoding, on my machine I always use encoding="utf-8"