pandas保存数据到csv_[pandas] 合并csv文件 + 打乱数据

最新推荐文章于 2023-06-01 08:00:00 发布

weixin_39621794

最新推荐文章于 2023-06-01 08:00:00 发布

阅读量307

点赞数

文章标签： pandas保存数据到csv

本文链接：https://blog.csdn.net/weixin_39621794/article/details/111665757

版权

data_财经.csv

data_房产.csv

data_家居.csv

# -*- coding: utf-8 -*-

import pandas as pd
import os
from sklearn.utils import shuffle



def get_labeled_data():
	data_1 = pd.read_csv('data_财经.csv')[:5]
	data_2 = pd.read_csv('data_房产.csv')[:5]
	data_3 = pd.read_csv('data_家居.csv')[:5]

	frames = [data_1, data_2, data_3]
	data = pd.concat(frames)
	data = shuffle(data) # 打乱
	data['id'] = range(len(data)) # id列
        # 去空格 换行符 
	data['content'] = data['content'].str.replace('n', '').replace(' ', '').replace('t', '').replace('r', '')
	data.set_index('id', inplace=True) # 设index
	data.to_csv('data1111.csv')
	
get_labeled_data()

data1111.csv