Python基本文档操作
读入文档:
f = open(‘test.txt’,’r’)
写入文档:
F = open(‘output.txt’,’w’)
注意!如果你想读文档,却写错了关键字,把’r’写成’w’,并运行了程序,那么你的文档就会变成空白的,因为你什么都没有写入。
修改文档里面指定行的内容:
比如你用代码实现了自动生成文本和符号,但是迫于格式要求,很多符号都是成对出现的,而且在生成完文本之后,很难再回撤,比如我就遇到这样的情况:
背景:
在yolo自动检测视频中的行人的时候,获取到了每一帧的帧序号,行人的坐标,而后这些坐标需要读入道VOTT(一款视频标注工具)里面进行细微矫正,所以需要转换成json格式,然后json格式..你懂的,中括号花括号,逗号都不能错,好说歹说我终于把自动转换格式的功能做出来了,也就是:从
到:
可以看到,我把每一帧里每一个行人的坐标都表示成json的格式了,但是格式转换之后有个问题,在倒数第三行的地方,多了个逗号:
不要看它只是一个小小的逗号,但是有这个逗号,VOTT就读不出来,没有逗号就读取成功,所以编程也是一份一丝不苟的工作呀~~
不就一个小逗号么?把它删了就OK啦~~
我也是这么想的,但是跟我交接的同事不这么认为,硬是要我把这个逗号自动去除掉,于是就有了今天的这次分享~
实现
首先创建两个文件读取函数f1和f2,f1负责读文件信息,f2写新文件信息,通过一读一写,实现指定内容的修改。首先f1读取这个json文档,然后一行一行地读,如果这一行不用改内容,就发直接写入新文件;如果这一行需要做内容更改,可以直接更改或者用文本替换的方式进行更改(为满足不同程序的自适应要求,多采用文本替换的方式),设定要替换的文本old_str和新文本new_str,进行内容替换,并完成其余行的文件读写。
源代码如下:
import os
file=’output.json’
old_str = '}],'
new_str = '}]'
#由于我只想把最后一个逗号去掉,所以新字符串比旧字符串少个逗号
#这里要确保只匹配到你想要更改的字符串,否则整一行符合条件的都会被替换掉
def alter(file,old_str,new_str):
total_count =0 #文档的总行数
count = 0 #目标行数
with open(file) as f:
for line in f:
total_count = total_count + 1
with open(file, "r", encoding="utf-8") as f1, \
open("%s.bak" % file, "w",encoding="utf-8") as f2:
for line in f1:
count = count+1
if count == total_count-2:如果当前行是倒数第三行,执行替换操作
if old_str in line:
line =line.replace(old_str, new_str)
f2.write(line)
else:
f2.write(line)
os.remove(file)
os.rename("%s.bak"% file, file)
好了,这样就愉快地实现了自动化的格式转换并直接能被VOTT识别啦~
最后放上一张VOTT载入YOLO检测的行人坐标的截图:
欢迎关注“pyhon修炼之道”,我们将持续更新新鲜python相关文章~