用简洁的语法写入文件

Question

有办法让下面的代码更紧凑吗？

header*是字符串变量。

output_file.write(header1) 
output_file.write("\n") 
output_file.write(header2) 
output_file.write("\n") 
output_file.write(header3) 
output_file.write("\n") 
output_file.write(header4) 
output_file.write("\n") 
output_file.write(header5) 
output_file.write("\n") 

Answer 1

看一看str.format：https://docs.python.org/3/library/stdtypes.html#str.format

你的情况：

"{}\n{}\n{}\n{}\n{}\n".format(
    header1, header2, header3, header4, header5 
) 

如果你喜欢headers = [header1, header2, header3, header4, header5]名单有他们，这将是更容易：

"{}\n{}\n{}\n{}\n{}\n".format(*headers) 

然后，你也可以使用str.join：

"\n".join(headers)