python读写各种格式文件

文章目录 前言1 读写txt2 读写xml文件3 读写json文件4 命名/拷贝移动文件总结

前言

 本篇博文主要记录一些常见的文件读取操作，不定期更新扩充。

---

1 读写txt import os # 读 def read_txt(path): with open(path,'r') as f: while True: line = f.readline() # 读取整行数据 if not line: break imgsets.append(line.strip()) # 必须放在 if 之后，否则会多读入一个空字符。 # 或者 lines = f.readlines() for line in lines: imgsets.append(line.strip()) #f.close() # f.close不用加，因为Python自动默认将操作系统的待写入的文件写到了磁盘里面。 # 写 def write_txt(path): with open(path,'w') as f: f.write('wulele\n') #f.close() # 和上面同理，with方法保证了 操作系统自动将文件完全写入磁盘了。 if __name__ == '__main__': # read read_path = '/home/wujian/MMLAB/gallery_content.txt' read_txt(read_path) # write write_path = '/home/wujian/MMLAB/save.txt' write_txt(write_path) 2 读写xml文件 import xml.etree.ElementTree as ET def save_to_new_annotations(img,path): doc = ET.parse(path) root = doc.getroot() width,height,_ = img.shape for size in root.findall('size'): size.find('width').text = str(width) size.find('height').text = str(height) f = 'E:/RotateVocDataset/VOCdevkit/VOC2007/NewAnnotations/000005.xml' doc.write(f,encoding="utf-8") #重新写入参数文件 print('Done!') def read_annotation(path): root = ET.parse(path).getroot() sizes = [] filename = root.find('filename').text # text后无括号 # 获取图像的尺寸 for size in root.findall('size'): width = size.find('width').text height = size.find('height').text channel = size.find('depth').text sizes.extend([width,height,channel]) # 获取物体信息 imdbs = [] for obj in root.findall('object'): imdb = [] label = obj.find('name').text bbox = obj.find('bndbox') # 由于bndbox非叶子节点，所以不能直接.text xmin = bbox.find('xmin').text # 获取bbox的信息 ymin = bbox.find('ymin').text xmax = bbox.find('xmax').text ymax = bbox.find('ymax').text imdb.extend([xmin,ymin,xmax,ymax,label]) imdbs.append(imdb) return sizes,imdbs if __name__ == '__main__': img_path= 'E://RotateVocDataset/VOCdevkit/VOC2007/JPEGImages/000001.jpg' anno_path = 'E://RotateVocDataset/VOCdevkit/VOC2007/Annotations/000001.xml' img=cv2.imread(img_path) height, width ,_ = img.shape size, imdbs = read_annotation(anno_path) print(size,'\n',imdbs) 3 读写json文件

 json最大好处就是能够存储python的数据结构，即写进去是dict，读出来依旧是dict，所以大多数数据集都喜欢以json进行保存。而其中用到的主要api是：json.dumps(), json.loads()。

root_dir = '/mnt/data/VOC2017/train/' sjs = root_dir + 'coco_hoi_train2017.json' # 写入json with open(sjs, 'w') as f: #f.write(str(annot) + '\n') # 写的方法会破坏python的数据结构 f.write(json.dumps(content) + '\n') # json.dumps 就能保存python的数据结构 # 读取json with open(json_path,'r') as f: lines = f.readlines() for line in lines: content = json.loads(line.strip()) print(content['img'], content['label']) break 4 命名/拷贝移动文件 filename.zfill(6) # 字符串填充6位，右对齐。 shutile.copyfile 总结

 没啥可总结的…