python - 写入文件时，您能否具体说明要写入的位置？

Question

我有一个文本文件，其中包含以下内容:

    20
    15
    10

我有以下代码:

    test_file = open("test.txt","r")
    n = 21
    line1 = test_file.readline(1)
    line2 = test_file.readline(2)
    line3 = test_file.readline(3)
    test_file.close()
    line1 = int(line1)
    line2 = int(line2)
    line3 = int(line3)
    test_file = open("test.txt","a")
    if n > line1:
        test_file.write("\n")
        n = str(n)
        test_file.write(n)
        test_file.close()

此代码检查变量“n”是否大于第 1 行。我想要它做的是如果它大于第 1 行，则应将其写在前一行 1 之前的一行中。但是此代码将把它写在文件的底部。我可以做些什么来在我想写的地方而不是在文件底部写一些东西吗？感谢您的帮助。

Answer 1

您可以将所有数据放在一个变量中，编辑该变量，然后覆盖文件中的信息。

with open('test.txt', 'r') as file:
    # read a list of lines into data
    data = file.readlines()

# now change the 2nd line, note that you have to add a newline
data[1] = "42\t\n" 

# and write everything back
with open('test.txt', 'w') as file:
    file.writelines( data )

这是一个简短的回答，实现你自己的算法来解决你自己的问题。