Question

我正在尝试找到“ AAXX”并在上面两行添加单词“ Hello”：

Input:
111
222
AAXX
333
444
AAXX
555
666
AAXX

Output: 
Hello 
111
222
AAXX
Hello
333 
444
AAXX
Hello
555
666
AAXX

通过使用下面的代码，我设法仅在第一个“ AAXX”之前的两行插入一个“ Hello”，但是我无法使其在文件中循环并且对所有“ AAXX”匹配都执行相同的操作。

import os

with open(os.path.expanduser("~/Desktop/test.txt"), "r+") as f:
    a = [x.rstrip() for x in f]
    for i, item in enumerate(a):
        if item.startswith("AAXX"):
            a.insert(i-2,"Hello")
            break
        index += 1
    # Go to start of file and clear it
    f.seek(0)
    f.truncate()
    # Write each line back
    for line in a:
        f.write(line + "\n")

到目前为止，我得到了：

Hello
111
222
AAXX
333
444
AAXX
555
666
AAXX

Answer 1

def p(a):
    r = []
    for i, item in enumerate(a):
        if item.startswith("AAXX"):
            r.append(i)
    for i in reversed(r):
        a.insert(i-2,"HELLO")
    return(a)

您可以将其包装起来，以处理输入/输出。您需要解决在前两个项目中出现“ AAXX”的情况，因为您还没有定义想要的行为。关键问题在于，在迭代过程中修改列表是一种不好的做法，尤其是因为您插入了较早的“ HELLO”，所以以后的索引可能会关闭。一种可能的解决方案是跟踪所有插入索引，然后按反向顺序进行插入，因为在列表中的后面插入不会影响较早的索引。

Answer 2

您可以尝试以下方法吗？

with open('test.txt', 'r') as infile:
    data = infile.read()
final_list = []
for ind, val in enumerate(data.split('\n')):
    final_list.append(val)
    if val == 'AAXX':
        final_list.insert(-3, 'HELLO')
# save the text file
with open('test.txt', 'w') as outfile:
    data = outfile.write('\n'.join(final_list))

输出：

HELLO
111
222
AAXX
HELLO
333
444
AAXX
HELLO
555
666
AAXX

在python中的匹配之前添加文本2行

2 个答案: