使用pandas以.txt格式保存拆分数据集

Question

尝试将数据集吐槽到train和test，然后需要将其保存为.txt格式。

这是到目前为止的代码,

import pandas as pd
from sklearn.model_selection import train_test_split

category=pd.read_csv('dataset.tsv',delimiter='\t',encoding='utf-8')

train, test = train_test_split(category, test_size=0.2)

test.to_csv('checkme.txt')

Answer 1

您需要将数据帧编写为Unicode：

test.to_csv('checkme.txt', sep='\t', encoding='utf-8')