python - 如何使用python在文件的第一行之前插入一个新行？

Question

更多详情如下:

1st line

2nd line

3rd line

4th line

...

现在想在 1st line 之前插入一个名为 zero line 的新行。文件如下所示:

zero line

1st line

2nd line

3rd line

4th line

...

我知道 sed 命令可以完成这项工作，但是如何使用 python 完成呢？谢谢

Answer 1

你可以使用fileinput

>>> import fileinput
>>> for linenum,line in enumerate( fileinput.FileInput("file",inplace=1) ):
...   if linenum==0 :
...     print "new line"
...     print line.rstrip()
...   else:
...     print line.rstrip()
...