python - Python ElementTree：部分解析大文件

Question

我有一个大的 XML 文件，它的结构大致为（按此顺序）：

<document>
   <interesting_part>
     ...
   </interesting_part>
   <foo>
     ...
     60000 lines
     ...
   </foo>
</document>

我的程序是：

from xml.etree import ElementTree as et
f=open(path_f)
tree=et.parse(f)
f.close()
# retreive infos from tree...

只有前几个块让我对文件感兴趣，但性能很低，因为 et.parse() 加载了整个文件。

如何加载文件只到 </interesting_part> ？

我想到了类似的东西：

class My_Parser(et.XMLParser):
    ????
my_parser = My_Parser()
tree=et.parse(f, my_parser)

提前谢谢你，埃里克。

Answer 1

请改用该iterparse()函数，并在您拥有所需内容时停止迭代：

for event, element in et.iterparse(f):
    if element.tag == 'interesting_part':
        # `element` is the complete <interesting_part> element, with children
        # process it
        break  # ends parsing.