numpy 在文件中不同行的末尾添加不同的数据

whlutmcx  于 2023-01-26  发布在  其他
关注(0)|答案(2)|浏览(180)

我有一个具有以下结构的txt文件:

我还想在每一个长行的末尾添加它们上面的短行的数据(逗号后面),不带描述(STN_NO,STN_ID,INST_HT),如下所示:

有可能吗?有什么想法吗?
另外,我使用的是Python 3.3版。

puruo6ea

puruo6ea1#

或者,您可以使用不涉及正则表达式的更简单(虽然更长)的解决方案。

f = open('file.txt')
for line in f:
    line = line.replace('\n', '')
    if 'STN_NO' in line:
        stn_no = line.split(',')[-1]
        print(line)
    elif 'STN_ID' in line:
        stn_id = line.split(',')[-1]
        print(line)
    elif 'INST_HT' in line:
        inst_ht = line.split(',')[-1]
        print(line)
    else:
        print(line[:-1] + ',' + stn_no + ',' + stn_id + ',' + inst_ht)

注意,这会将INST_HT行中的分号放回到每一个长行的末尾。如果不需要,可以使用inst_ht[:-1]将其删除。

elcex8rz

elcex8rz2#

让我们假设您的映像中的文件为以下简化版本:

STN_NO, 41943043
STN_ID, KAST
INST_HT, 1.01500;
Line 1
Line 2
Line 3
STN_NO, 41943062
STN_ID, S2
INST_HT, 0.75;
Line 4
Line 5
Line 6
STN_NO, 123456
STN_ID, XXX
INST_HT, 0.99;
Line 7
Line 8
Line 9

您可以使用正则表达式来捕获块中的模式并合并:

import re

pat=re.compile(r'^STN_NO,\s+([^\n]+)$\s*^STN_ID,\s+([^\n]+)$\s*^INST_HT,\s+([^;]+);\s*(.*?)(?=^STN_NO|\Z)', re.S | re.M)

with open(fn) as f:
    txt=f.read()

for mg in pat.finditer(txt):
    for line in mg.group(4).splitlines():
        print(line+','+','.join([mg.group(1), mg.group(2), mg.group(3)]))

图纸:

Line 1,41943043,KAST,1.01500
Line 2,41943043,KAST,1.01500
Line 3,41943043,KAST,1.01500
Line 4,41943062,S2,0.75
Line 5,41943062,S2,0.75
Line 6,41943062,S2,0.75
Line 7,123456,XXX,0.99
Line 8,123456,XXX,0.99
Line 9,123456,XXX,0.99

如果您的文件大于内存所能容纳的大小,请使用mmap进行虚拟化。

相关问题