python利用lxml读写xml格式的文件


Posted in Python onAugust 10, 2017

之前在转换数据集格式的时候需要将json转换到xml文件,用lxml包进行操作非常方便。

1. 写xml文件

a) 用etree和objectify

from lxml import etree, objectify

E = objectify.ElementMaker(annotate=False)
anno_tree = E.annotation(
  E.folder('VOC2014_instance'),
  E.filename("test.jpg"),
  E.source(
    E.database('COCO'),
    E.annotation('COCO'),
    E.image('COCO'),
    E.url("http://test.jpg")
  ),
  E.size(
    E.width(800),
    E.height(600),
    E.depth(3)
  ),
  E.segmented(0),
)

etree.ElementTree(anno_tree).write("text.xml", pretty_print=True)

输出的test.xml文件内容如下:

```

如果需要在anno_tree的基础上加其他标签的话用append即可:

E2 = objectify.ElementMaker(annotate=False)
anno_tree2 = E2.object(
  E.name("person"),
  E.bndbox(
    E.xmin(100),
    E.ymin(200),
    E.xmax(300),
    E.ymax(400)
  ),
  E.difficult(0)
)
anno_tree.append(anno_tree2)

上面的输出就变成了:

<annotation>
 <folder>VOC2014_instance/person</folder>
 <filename>test.jpg</filename>
 <source>
  <database>COCO</database>
  <annotation>COCO</annotation>
  <image>COCO</image>
  <url>http://test.jpg</url>
 </source>
 <size>
  <width>800</width>
  <height>600</height>
  <depth>3</depth>
 </size>
 <segmented>0</segmented>
 <object>
  <name>person</name>
  <bndbox>
   <xmin>100</xmin>
   <ymin>200</ymin>
   <xmax>300</xmax>
   <ymax>400</ymax>
  </bndbox>
  <difficult>0</difficult>
 </object>
</annotation>

b) 用etree和SubElement

annotation = etree.Element("annotation")
etree.SubElement(annotation, "folder").text = "VOC2014_instance"
etree.SubElement(annotation, "filename").text = "test.jpg"
source = etree.SubElement(annotation, "source")
etree.SubElement(source, "database").text = "COCO"
etree.SubElement(source, "annotation").text = "COCO"
etree.SubElement(source, "image").text = "COCO"
etree.SubElement(source, "url").text = "http://test.jpg"
size = etree.SubElement(annotation, "size")
etree.SubElement(size, "width").text ='800' # 必须用string
etree.SubElement(size, "height").text = '600'
etree.SubElement(size, "depth").text = '3'
etree.SubElement(annotation, "segmented").text = '0'
key_object = etree.SubElement(annotation, "object")
etree.SubElement(key_object, "name").text = “person”
bndbox = etree.SubElement(key_object, "bndbox")
etree.SubElement(bndbox, "xmin").text = str(100)
etree.SubElement(bndbox, "ymin").text = str(200)
etree.SubElement(bndbox, "xmax").text = str(300)
etree.SubElement(bndbox, "ymax").text = str(400)
etree.SubElement(key_object, "difficult").text = '0'
doc = etree.ElementTree(annotation)
doc.write(open("test.xml", "w"), pretty_print=True)

2. 读xml

这里可以用xpath直接提取所需的元素的值。比如想要获取上面test.xml文件的x, y坐标:

tree = etree.parse("test.xml")
# get bbox
for bbox in tree.xpath('//bndbox'):  # 获取bndbox元素的内容
  for corner in bbox.getchildren(): # 便利bndbox元素下的子元素
    print corner.text  # string类型

参考

http://lxml.de/tutorial.html

https://stackoverflow.com/questions/12657043/parse-xml-with-lxml-extract-element-value

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python列表推导式的使用方法
Nov 21 Python
Python中的引用和拷贝浅析
Nov 22 Python
pygame学习笔记(6):完成一个简单的游戏
Apr 15 Python
简单了解Python下用于监视文件系统的pyinotify包
Nov 13 Python
Flask框架踩坑之ajax跨域请求实现
Feb 22 Python
Python参数解析模块sys、getopt、argparse使用与对比分析
Apr 02 Python
Python循环中else,break和continue的用法实例详解
Jul 11 Python
flask框架jinja2模板与模板继承实例分析
Aug 01 Python
Python:合并两个numpy矩阵的实现
Dec 02 Python
如何使用python传入不确定个数参数
Feb 18 Python
python GUI库图形界面开发之PyQt5线程类QThread详细使用方法
Feb 26 Python
Python之京东商品秒杀的实现示例
Jan 06 Python
Python编程实现及时获取新邮件的方法示例
Aug 10 #Python
Python中函数eval和ast.literal_eval的区别详解
Aug 10 #Python
Python基础之getpass模块详细介绍
Aug 10 #Python
Python中字典(dict)合并的四种方法总结
Aug 10 #Python
详解Python 模拟实现生产者消费者模式的实例
Aug 10 #Python
Python 操作文件的基本方法总结
Aug 10 #Python
Python 模拟登陆的两种实现方法
Aug 10 #Python
You might like
PHP初学者最感迷茫的问题小结
2010/03/27 PHP
在PHP中设置、使用、删除Cookie的解决方法
2013/05/06 PHP
PHP开发框架laravel安装与配置教程
2015/03/13 PHP
php简单实现无限分类树形列表的方法
2015/03/27 PHP
PHP实现微信提现功能(微信商城)
2019/11/21 PHP
PHP论坛实现积分系统的思路代码详解
2020/06/01 PHP
javascript Select标记中options操作方法集合
2008/10/22 Javascript
(跨浏览器基础事件/浏览器检测/判断浏览器)经验代码分享
2013/01/24 Javascript
简略说明Javascript中的= =(等于)与= = =(全等于)区别
2013/04/16 Javascript
js动态修改input输入框的type属性(实现方法解析)
2013/11/13 Javascript
实现前后端数据交互方法汇总
2015/04/07 Javascript
JavaScript实现仿淘宝商品购买数量的增减效果
2016/01/22 Javascript
jQuery Timelinr实现垂直水平时间轴插件(附源码下载)
2016/02/16 Javascript
bootstrap输入框组代码分享
2016/06/07 Javascript
手机端实现Bootstrap简单图片轮播效果
2016/10/13 Javascript
Node.js中文件操作模块File System的详细介绍
2017/01/05 Javascript
JS中数组与对象的遍历方法实例小结
2018/08/14 Javascript
layer实现弹出层自动调节位置
2019/09/05 Javascript
[01:19:34]2014 DOTA2国际邀请赛中国区预选赛 New Element VS Dream time
2014/05/22 DOTA
Python实现获取操作系统版本信息方法
2015/04/08 Python
在Django中实现添加user到group并查看
2019/11/18 Python
Python基于network模块制作电影人物关系图
2020/06/19 Python
css3的transform造成z-index无效解决方案
2014/12/04 HTML / CSS
html5 canvas绘制网络字体的常用方法
2019/08/26 HTML / CSS
英格兰橄榄球商店:England Rugby Store
2016/12/17 全球购物
英国的屈臣氏:Boots博姿
2017/12/23 全球购物
Lacoste(法国鳄鱼)加拿大官网:以标志性的POLO衫而闻名
2019/05/15 全球购物
SEPHORA丝芙兰德国官方购物网站:化妆品、护肤品和香水
2020/01/21 全球购物
Super-Pharm波兰:药房和香水在一个地方
2020/08/18 全球购物
家长给孩子的评语
2014/01/30 职场文书
幼儿园老师寄语
2014/04/03 职场文书
物流专业求职信
2014/06/30 职场文书
三月法制宣传月活动总结
2014/07/03 职场文书
实验心得体会范文
2016/01/25 职场文书
CentOS下安装Jenkins的完整步骤
2022/04/07 Servers
Python实现将多张图片合成MP4视频并加入背景音乐
2022/04/28 Python