使用Pandas分块处理大文件
使用Pandas分块处理大文件 问题:今天在处理快手的用户数据时,遇到了一个差不多600M的txt文本,用sublime打开都蹦了,我用pandas.read_table()去读,差不多花了近2分钟,最后打开发现差不多3千万行数据。这仅仅是打开,如果要处理不知得多费劲。 解决:我翻了一下文档,这一类读取文件的函数有两个参数:chunksize、iterator 原理就是不一次性把文件数据读入内存中,而是分多次。 1、指定chunksize分块读取文件 read_csv 和 read_table 有一个 chunksize 参数,用以指定一个块大小(每次读取多少行),返回一个可迭代的 TextFileReader 对象。 table=pd.read_table(path+'kuaishou.txt',sep='\t',chunksize=1000000) for df in table: 对df处理 #如df.drop(columns=['page','video_id'],axis=1,inplace=True) #print(type(df),df.shape)打印看一下信息12345 我这里又对文件进行了划分,分成若干个子文件分别处理(没错,to_csv也同样有chunksize参数) 2、指定iterator=True iterator