python - 将一个文件的文本插入到另一个文件的特定行

Question

我有两个文本文件 test1.txt 和 test2.txt。我希望能够读取 test1.txt 内容并将其写入特定行的 text2.txt。

我已经能够读取 test1.txt 的内容并将它们写入空白 test2.txt。

现在我想更改我的代码以读取 test1.txt 的上下文并将它们写入非空的 text2.txt。

with open("/test1.txt") as f:
    lines = f.readlines()

with open("/test2.txt", "w") as f1:
    f1.writelines(lines)

如何将 test1.txt 的内容插入到 test2.txt 的第 20 行。

Answer 1

简单但有点占用内存:

# insert contents of "/test1.txt" into "/test2.txt" at line 20
with open("/test1.txt", "r") as f1:
    t1 = f1.readlines()
with open("/test2.txt", "r") as f2:
    t2 = f2.readlines()
t2.insert(20, t1)
with open("/test2.txt", "w") as f2:
    f2.writelines(t2)

最小化内存使用和磁盘写入，但有点复杂:

# insert contents of "/test1.txt" into "/test2.txt" at line 20
with open("/test2.txt", "rw+") as f2:
    for x in range(20):
        f2.readline()   # skip past early lines
    pos = f2.tell() # remember insertion position
    f2_remainder = f2.read()    # cache the rest of f2
    f2.seek(pos)
    with open("/test1.txt", "r") as f1:
        f2.write(f1.read())
    f2.write(f2_remainder)

我还没有测试过其中任何一个，但它们应该可以帮助您完成大部分工作。