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

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)

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

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)

我还没有测试过这两种方法，但它们应该能让你大部分时间到达那里。

Answer 2

实际上两个已接受答案中的解决方案都已损坏（！）

这是seek和tell版本的固定版本：

with open("test2.txt", "r+") as f2:
    for x in range(20):
        f2.readline()
    pos = f2.tell()
    f2_remainder = f2.read()
    f2.seek(pos)
    with open("test1.txt", "r") as f1:
        for line in f1:
            f2.write(line)
    f2.write(f2_remainder)

如果输入文件很大并且您需要将其保存在内存中，它仍然会占用大量内存，但在现代计算机上，即使是千兆字节大小的文件，这也应该是可行的。

演示：https://ideone.com/UrOVmx（包括接受的答案如何错误的演示，以及 pavol.kutaj 答案的部署。）

Answer 3

这个问题可能有多种解决方案。一种可能的解决方案是。

1. 打开 file1 并将内容读取到列表中
2. 打开 file2 并将内容读取到另一个列表
3. 在 file2 中的所需位置添加 file1 的内容。
4. 在 file2 的第二个列表中写回内容。

Answer 4

试试这个，如果你想在特定行写入一个文件的数据，那么试试这个

fp = open("data.txt","r")
# fp is file1
data = fp.read()
f = open('workfile.txt', 'rw+')
#f is file2 on which you want to write data
ss = f.readlines()
f.seek(0)
f.truncate()
ss.insert(5,data)
#give your own line number instead of 5
f.write(''.join(ss))

Answer 5

如果您需要在包含特定子字符串第一次出现的行之后插入并且您不知道（不想依赖）确切的行号：

• 将您的数据放入source.txt（待充实）和insert.txt（待充实）中。

import json


def main():
    with open("source.txt", mode="rt", encoding="utf-8") as f1:
        source_list = [line for line in f1.readlines()]

    with open("insert.txt", mode="rt", encoding="utf-8") as f2:
        insert_list = [line for line in f2.readlines()]

    pos = [i for i, line in enumerate(source_list) if '<substring>' in line][0]
    source_list[pos: pos] = insert_list

    with open("source.txt", mode="wt", encoding="utf-8") as f1:
        f1.write("".join(source_list))


if __name__ == "__main__":
    main()