如何将 pandas.Dataframe 添加到创建为 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] Is a directory: './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"