python将100wcsv数据随机切分50份批量保存在文件夹中

努力成为一个帅气数据风控科学家

于 2022-08-01 19:21:50 发布

阅读量735

点赞数

分类专栏： python 文章标签： python 开发语言

本文链接：https://blog.csdn.net/weixin_43792309/article/details/126108010

版权

python 专栏收录该内容

19 篇文章 0 订阅

订阅专栏


```python
import pandas as pd 
data = pd.read_csv("\\100w_split.csv",
                   encoding = 'gbk',# 注意要使用gbk 防止中文乱码
                   converters={'code':str} # 防止中文乱码，如果报错，删除本行
                   )
data['cust_id'] = data['cust_id'].astype('string')
data_col = ['核心客户号','label','row_no']# 将cust_id这个列名改为中文‘核心客户号’
data.columns = data_col
data = ''+ data['核心客户号'].astype(str) # ‘’+来避免我的客户号前导0消失

num = 0
for i in range(1,51):
    start = num
    num = num + int(data.shape[0]/50)
    file = '\t'+data.iloc[start:num,]
    file.to_csv("\\2q_100w\\"+str(i)+"_data.csv",
                index = False,#不要索引
                encoding = 'gbk')# 注意我们的列名是汉字 要用gbk格式