在数据处理业务中,经常要把文件结构相同或近似相同的数据文件合并成一个文件,或者将一个比较大的数据文件拆分成小的数据文件。本文将介绍文本文件和 Excel 文件合并及拆分会遇到的几种情况,并提供用 esProc SPL 编写的代码示例。esProc 是专业的数据计算引擎,SPL 中有完善的文件导入、导出及目录操作函数,非常适合做数据文件的合并及拆分工作。
一、 文件合并
1. 同构文本文件合并
在某个文件目录下有多个文本文件,这些文件表头和列结构完全相同,只是数据行数和数据内容不同,需要将这些文件的数据全部合并到一个文本文件中,共用同一个表头。
示例:在e:/orders目录下有每日的订单文本文件,每个文件的结构相同,第一行是列名,第二行开始是数据,如下图所示,请将它们合并成一个订单表文件orders.txt。
esProc SPL脚本如下:
A注释
1=directory@p("e:/orders/*.txt")返回orders目录所有txt文件的完整路径名
2=A1.conj(file(~).import@t())合并所有文件的数据
3=file("e:/orders.txt").export@t(A2)将合并后的数据写入orders.txt文件
如果e:/orders还有子目录,子目录中的txt文件也需要一起合并,那么把A1格改成=directory@ps("e:/orders/*.txt"),选项@s表示递归查找所有子目录下的文件。
上面这段脚本是假设内存能够装下合并以后的全部数据,如果装不下,那么应该用下面这段脚本:
ABC
1=directory@p("e:/orders/*.txt")=file("e:/orders.txt")
2for A1=file(A2).cursor@t()
3if #A2==1 =B1.export@t(B2)
4else=B1.export@a(B2)
A1列出目录中的所有txt文件的完整路径名称,如果要查找子目录,则加上@s选项
A2对列出的文件进行循环
B2用游标读取每一个文件,@t表示第一行是列名
B3-C4将B2游标中数据导出,第一次要导出列名,其后用@a进行追加
2. 结构近似的文本文件合并
如果文件结构并不是完全相同,比如列的顺序不一样、列数不一样,但各文件都含有共同的几列,想要把这些共同列的数据都合并到一个文件中。合并这些文件时,需要按指定顺序读出每个文件中的这些共同列数据。
示例:还是上面这个例子,已知所有订单文件都有ID、Company、Area、OrderDate、Amount这5列,但各文件中列的顺序并不相同,有的文件还有其它一些列,请将各文件中的这5列数据合并到orders.txt文件中。
esProc SPL脚本如下:
A注释
1=directory@p("e:/orders/*.txt")返回orders目录所有txt文件的完整路径名
2=A1.conj(file(~).import@t(ID,Company,Area,OrderDate,Amount))按指定的5列顺序读出各文件数据,然后合并所有文件的数据
3=file("e:/orders.txt").export@t(A2)将合并后的数据写入orders.txt文件
同样地,如果内存装不下合并后的所有数据,则使用下面这段脚本:
ABC
1=directory@p("e:/orders/*.txt")=file("e:/orders.txt")
2for A1=file(A2).cursor@t(ID,Company,Area,OrderDate,Amount)
3if #A2==1 =B1.export@t(B2)
4else=B1.export@a(B2)
3. 文件名转成列数据
在合并数据的同时,需要为合并后的数据增加一列,并用合并前的文件名给此列赋值,以便标记数据类别或来源。
示例:在e:/orders目录下有各种零件的订单Excel文件,文件名就是零件的名字,比如tyre.xlsx、engine.xlsx……等,每个文件的结构相同,第一行是列名,第二行开始是数据,如下图所示,请将它们合并成一个订单表文件orders.xlsx,并增加一列PartName用来记录零件的名字。
esProc SPL脚本如下:
A注释
1=directory@p("e:/orders/*.xlsx")返回orders目录所有Excel文件的完整路径名
2=A1.conj((fn=filename@n(~),file(~).xlsimport@t().derive(fn:PartName)))先将文件名定义为临时变量fn,读入文件后,增加一列PartName,并用fn赋值。最后将各文件数据合并
3=file("e:/orders.xlsx").xlsexport@t(A2)将合并后的数据写入orders.xlsx文件
同样地,如果内存装不下合并后的所有数据,则使用下面这段脚本:
ABC
1=directory@p("e:/orders/*.xlsx")=file("e:/orders.xlsx")
2for A1=file(A2).xlsimport@tc().derive(filename@n(A2):PartName)
3if #A2==1=B1.xlsexport@ts(B2)
4else=B1.xlsexport@as(B2)
合并后的orders.xlsx文件部分数据如下图所示:
二、 文件拆分
1. 分组拆分
对文件中数据进行分组,把每组数据单独存为一个文件,用组名为文件命名。
示例:在订单表Excel文件中有各种零件的订单,请把同种零件的订单各存为一个Excel文件,以便发送给零件生产部门。
esProc SPL脚本如下:
AB
1=file(“e:/orders/orders.xlsx”).xlsimport@t() =A1.group(partName)
2for B1=file("e:/parts/"+A2(1).partName+”.xlsx”).xlsexport@t(A2)
A1读入所有原始数据
B1按partName分组
A2循环每个零件组的订单信息
B2以零件名称作为文件名,导出零件信息
如果原文件很大,不能全部装入内存,那么应该使用游标方式读数,脚本如下:
ABCD
1=file(“e:/orders/orders.xlsx”).xlsimport@tc()
2for A1,50000=A2.group(partName)
3for B2=file("e:/parts/"+B3(1).partName+”.xlsx”)
4if C3.exists()=C3.xlsexport@a(B3)
5else=C3.xlsexport@t(B3)
A1创建游标读取原始数据
A2循环游标读数,每次读50000行(读多少行根据内存大小决定)
B2对每次读取的数据按partName分组
B3循环每组零件
C3以零件名称作为文件名创建文件对象
C4-D5如果文件已存在,则用@a追加写入零件订单信息,不存在则用@t先写入列名再导入数据
2. 记录占据多行的拆分
在文本文件中,一条数据记录是由多行数据组成的,在拆分成小文件时,要识别哪几行是同一条数据记录,保证同一条数据记录不会被拆分到两个文件中。
示例1:有网站运行日志文件log.txt,每条日志由5行组成,现在需要把这个日志文件拆分成一些小文件,每个文件由1000条日志组成。
esProc SPL脚本如下:
AB
1=file(“e:/log.txt”:”UTF-8”).cursor@s()
2for A1,5000=file(“e:/log/log_”/#A2/”.txt”).export(A2)
A1用游标读取日志文件数据,@s表示将整行读成一个字符串
A2循环游标,每次取5000行,刚好是1000条日志
B2按循环序号生成日志文件名,如log_1.txt、log_2.txt……,然后将当前取出的所有行写入
示例2:有程序运行日志文件log.txt如下图所示,每条日志由不确定的几行组成,每条日志由中括号开头,其后只要不是中括号开头的行,都与它属于同一条日志。现在需要把这个日志文件拆分成一些小文件,每个文件由1000条日志组成。
esProc SPL脚本如下:
AB
1=file(“e:/log.txt”:”UTF-8”).read@n() =A1.select(~!="")
2=B1.group@i(pos(~,"[")==1).cursor()
3for A2,1000=file(“e:/log/log_”/#A3/”.txt”)
4=A3.(~.(B3.write@a(~)))
A1打开日志文件读取数据,@n表示将每一行读成一个字符串,所有串组成一个序列返回
B1筛选出非空的行
A2按行是否用中括号开头作为分组条件,中括号开头的作为一个新组,不是则并到当前组。最后把所有组序列转换成游标
A3循环游标,每次取1000个组,即1000条日志
B3按循环序号生成日志文件名,如log_1.txt、log_2.txt……
B4两层循环,外层是循环1000个组,内层循环每组的成员(即数据行),将每行追加写入文件
《SPL CookBook》中有更多相关计算示例。