Python实现大文本文件按行数切割方案
针对产品功能限制导致无法直接处理特大文件的问题,本文提供了一种基于Python的自定义大文本文件切割方案。该方案支持CSV及TXT等格式,按指定行数将大文件拆分为多个带列头的小文件。通过Python 3.6环境下的实际测试,1.4GB CSV文件分割耗时55秒,展示了较高的开发效率与处理性能。
在实际业务场景中,受限于现有产品的功能约束,系统往往无法直接处理超大型文件,因此必须对大文件进行切割处理。
虽然可以使用UltraEdit等编辑工具,或从网络下载专用的文件切割软件,但这些方法要么依赖手工操作且流程繁琐,要么无法灵活满足特定的自定义需求。
对于开发人员而言,自主开发一个专用的处理工具(即“造轮子”)通常是必要且合理的。
Python作为一种高效的快速开发语言,具备强大的代码表达能力和较高的开发效率,因此使用Python快速实现该功能是完全可行的。
具体需求如下:
输入源为包含列头的CSV文件、TXT文件或其他任意文本文件。
输出结果为一系列带有明确标识的小文件,每个小文件内部包含指定数量的行数。
开发环境配置为Python 3.6。
核心实现代码如下:
# -*- coding: cp936 -*-
import os
import time
def mkSubFile(lines,head,srcName,sub):
[des_filename, extname] = os.path.splitext(srcName)
filename = des_filename + '_' + str(sub) + extname
print( 'make file: %s' %filename)
fout = open(filename,'w')
try:
fout.writelines([head])
fout.writelines(lines)
return sub + 1
finally:
fout.close()
def splitByLineCount(filename,count):
fin = open(filename,'r')
try:
head = fin.readline()
buf = []
sub = 1
for line in fin:
buf.append(line)
if len(buf) == count:
sub = mkSubFile(buf,head,filename,sub)
buf = []
if len(buf) != 0:
sub = mkSubFile(buf,head,filename,sub)
finally:
fin.close()
if __name__ == '__main__':
begin = time.time()
splitByLineCount('盂县.csv',600000)
end = time.time()
print('time is %d seconds ' % (end - begin))
测试结果显示:
对大小为1.4GB的CSV文件(包含13列数据)进行分割,生成23个小文件,总耗时55秒。
以上即为利用Python实现大文本文件切割的完整方案,旨在为读者提供参考,欢迎进一步交流与探讨。
免责声明:本文内容仅供参考,不代表本站观点。转载或引用请注明出处。









