Python - 如何使用 xml.etree.ElementTree 迭代每个 xml 节点返回列表？

Question

我正在使用 xml.etree.ElementTree 模块来解析 XML 文件，将属性返回到列表中，然后将这些列表输入 MySQL 数据库中(这最后一步我并不担心，因此无需涵盖在这里)。很简单，我目前能够这样做，但一次只能针对一个子节点。目标是对多个子节点执行此操作，无论有多少个子节点。这是一个示例文件:

<?xml version="1.0"?>
    <catalog>
       <book id="bk101" type="hardcover">
          <info author="Gambardella, Matthew" title="XML Developer's Guide" genre="Computer" price="44.95" publish_date="2000-10-01" description="An in-depth look at creating applications 
          with XML." />
       </book>
       <book id="bk102" type="softcover">
          <info author="Ralls, Kim" title="Midnight Rain" genre="Fantasy" price="5.95" publish_date="2000-10-01" description="A former architect battles corporate zombies, 
          an evil sorceress, and her own childhood to become queen 
          of the world." />
       </book>
       <book id="bk101" type="softcover">
          <info author="Corets, Eva" title="Maeve Ascendant" genre="Fantasy" price="5.95" publish_date="2000-11-17" description="After the collapse of a nanotechnology 
          society in England, the young survivors lay the 
          foundation for a new society." />
       </book>
    </catalog>

我能够通过返回具有正确属性的列表来解析 id="bk101"的第一个图书节点或 id="bk103"的最后一个图书节点的正确属性。但是，当我需要返回多个列表(每个图书节点和信息节点一个，因此在本例中总共 6 个列表)时，我只为每个文件返回一个列表。

这是我的代码:

import xml.etree.ElementTree

book_attribute = ['id', 'type']
info_attribute = ['author', 'title', 'genre', 'price', 'publish_date', 'description']


class ApplicationClass(object):  # define the only class in this file
    def __init__(self):
        self.ET = xml.etree.ElementTree.parse('file.xml').getroot()
        self.bookNodes = self.ET.findall('book')
        self.book_values_list = []
        self.info_values_list = []

    def get_book(self):
        for bookNode in self.bookNodes:
            self.book_values_list = [bookNode.get(i) for i in book_attribute]
        return self.book_values_list

    def get_info(self):
        for bookNode in self.bookNodes:
            for infoNode in bookNode.findall('info'):
                self.info_values_list = [infoNode.get(i) for i in info_attribute]
        return self.info_values_list

a = ApplicationClass()
a.get_book()
print(a.book_values_list)
a.get_info()
print(a.info_values_list)

所以我知道我的问题是每个函数只返回一个列表，因为我在函数末尾返回列表，然后在脚本末尾调用该函数。我只是找不到实现我想要的结果的正确方法。如果我不在脚本末尾运行我的函数，那么如何返回我正在查找的多个列表？

Answer 1

这一行是你的问题:

self.book_values_list = [bookNode.get(i) for i in book_attribute]

该行将用新列表替换现有列表。但是这行代码位于循环内，这意味着每次通过循环时，您都会丢失之前处理的内容。

我认为你想要这个:

self.book_values_list.append([bookNode.get(i) for i in book_attribute])

使用 .append() 而不是 = 将使您的变量中插入更多内容。最终你会得到一个列表列表，如下所示:

[['bk101', 'hardcover'], ['bk102', 'softcover'], ['bk101', 'softcover']]

您的其他方法/循环也有同样的问题 - 您将一个新列表分配给变量，而不是将新列表插入到现有列表中。