Python - 文件 I/O 代码中的列表理解

Question

我想使用列表理解来复制以下代码的功能:

with open('file.txt', 'w') as textfile:
    for i in range(1, 6):
        textfile.write(str(i) + '\n')

我尝试了以下方法:

with open('file.txt', 'w') as textfile:
    textfile.write(str([i for i in range(1, 6)]) + '\n')

但它(可以理解地)打印 [1, 2, 3, 4, 5]，而不是一行中的一个数字。

对于“您为什么要这样做？”我没有答案；我只是想看看这是否可能。谢谢!

编辑:谢谢大家的回复；出于某种原因，我的印象是列表理解总是封装在 [] 中。

Answer 1

一种方法是 file.writelines() :

with open('file.txt', 'w') as textfile:
    textfile.writelines(str(i) + "\n" for i in range(1, 6))