python提供了read,write,但和很多语言类似似乎没有提供insert。当然真要提供的话,肯定是可以实现的,但可能引入insert会带来很多其他问题,比如在插入过程中crash掉可能会导致后面的内容没来得及写回。
不过用fileinput可以简单实现在特定行插入的需求:
Python代码
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
|
import os
import fileinput
def file_insert(fname,linenos = [],strings = []):
"""
Insert several strings to lines with linenos repectively.
The elements in linenos must be in increasing order and len(strings)
must be equal to or less than len(linenos).
The extra lines ( if len(linenos)> len(strings)) will be inserted
with blank line.
"""
if os.path.exists(fname):
lineno = 0
i = 0
for line in fileinput. input (fname,inplace = 1 ):
# inplace must be set to 1
# it will redirect stdout to the input file
lineno + = 1
line = line.strip()
if i< len (linenos) and linenos[i] = = lineno:
if i> = len (strings):
print "\n" ,line
else :
print strings[i]
print line
i + = 1
else :
print line
file_insert( 'a.txt' ,[ 1 , 4 , 5 ],[ 'insert1' , 'insert4' ])
|
其中需要注意的是 fileinput.input的inplace必须要设为1,以便让stdout被重定向到输入文件里。
当然用fileinput.input可以不仅用来在某行插入,还可以在特定模式的行(比如以salary:结尾的行)插入或替换,实现一个小型的sed。
以上就是python文件特定行插入和替换的简单实例,如果大家有不明白或者好的建议请到留言区或者社区提问和交流,使用感谢阅读,希望能帮助到大家,谢谢大家对本站的支持!
原文链接:http://ryanstd.iteye.com/blog/480781