0

我正在编写读取顺序 ascii 文件并对其进行更改的代码。

到目前为止，我已经成功地读取了文件并替换了字符串。例如：

import os
import re
parent_folder_path = '.'
for eachFile in os.listdir(parent_folder_path):
    if eachFile.endswith('.sos'):
        newfilePath = parent_folder_path+'/'+eachFile
        file = open(newfilePath, 'r')
        sos = file.read()
        file.close()
        sos = sos.replace('...KOORDSYS -1', '...KOORDSYS 22')
        file = open(newfilePath, 'w')
        file.write(str(sos))
        file.close()

但我希望找到一个特定的位置（即一个字符串）并在其后插入以下三个字符串。

.DEF

..DRIFTSMERKING T26

..SPENNING T16

我正在使用的顺序 ascii 文件的结构如下所示：

.HODE
..TEGNSETT ISO8859-1
..TRANSPAR
...KOORDSYS 22
...ORIGO-NØ 0 0
...ENHET 0.01
..OMRÅDE
...MIN-NØ 6593309 455619
...MAX-NØ 6729987 588458
..SOSI-VERSJON 4.0
..SOSI-NIVÅ 2
..DAT0 20130313

我想做的正是这样：

searchline = '..DATO'

lines = f.readlines()
i = lines.index(searchline)
lines.insert(i, '.DEF')
lines.insert(i+1, '..DRIFTSMERKING T26')
lines.insert(i+2, '..SPENNING T16')
4

1 回答 1

1

你很近。只需使用 for 循环来迭代输入行并应用您的两个规则。此示例添加数据但先查看文件是否已被转换。

import os
parent_folder_path = '.'
canned_insert = [
    '.DEF' + os.linesep,
    '..DRIFTSMERKING T26' + os.linesep,
    '..SPENNING T16' + os.linesep]
canned_insert_buf = ''.join(canned_insert)
for eachFile in os.listdir(parent_folder_path):
    if eachFile.endswith('.sos'):
        newfilePath = os.path.join(parent_folder_path, eachFile)
        lines = open(newfilePath, 'r').readlines()
        with open(newfilePath, 'w') as f:
            for i, line in enumerate(lines):
                if line.strip() == '...KOORDSYS -1':
                    f.write('...KOORDSYS 22' + os.linesep)
                else:
                    f.write(line)
                    if line.startswith('..DATO') and lines[i+1:i+4] != canned_insert:
                        f.write(canned_insert_buf)
于 2013-03-19T17:04:28.827 回答