python修改文件(fileinput)

时间:2021-10-18 01:19:19

上网查了好多方法都是将文件一行一行都进来,然后新开一个文件指针,将文件再一行行写进新文件里。
感觉不是太好,看到这个fileinput的库,发现大同小异,不过有一点就是可以在原文件上修改,不用新开文件。


举个简单的例子:
我有json文件1.json是如下格式:

/* 1 */
{
"_id" : "18372776",
"city" : "广州",
"gender" : "woman",
"birthday" : " 巨蟹座",
"is_vip" : 1,
"contribution" : 964,
"user_name" : "寶穎穎"
}

/* 2 */
{
"_id" : "133155049",
"city" : "广州",
"gender" : "woman",
"birthday" : " 狮子座",
"is_vip" : 0,
"contribution" : 42,
"user_name" : "鱼之头"
}

我想把文件里面的/* ... */换成一个,,可以通过以下的代码实现:

import fileinput

str1 = ','
for line in fileinput.input('1.json', inplace= True):
if line.startswith(r'/*'):
print str1.strip()
else:
print line.strip()

其中的strip()是为了防止写入多余的空行。
这样的操作很简单,只能整行操作,也有局限性,我没有具体去研究这个库,只是简单记录了一下…
That`s all.