![](/img/trans.png)
[英]How can i add just one row in dataframe when I convert from array to pandas.dataframe
[英]How can I add pandas.Dataframe to file that is created as test_data?
我像這樣分割訓練數據的數據和機器學習的測試數據
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'。 我應該怎么做?
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"
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.