在Python中将不同的数据类型从XML加载到字典中

Question

我使用 cElementTree 在循环中提取 xml 标签和值，然后将它们存储到字典中。

XML 文件包含：

<root>
    <tag1>['item1', 'item2']</tag1>
    <tag2>a normal string</tag2>
</root>

Python 代码（大致）：

import xml.etree.cElementTree as xml

xmldata = {}
xmlfile = xml.parse(XMLFile.xml)
for xmltag in xmlfile.iter():
    xmldata[xmltag.tag] = xmltag.text

我遇到的问题是 xml 文件包含不同的数据类型，其中包括 string 和 list。不幸的是，Element.text 将所有 xml 值保存为 string（包括列表）。

因此，当我从 XML 文件加载时，我有：

{'tag1':"['item1', 'item2']", 'tag2':'a normal string'}

当我希望有：

{'tag1':['item1', 'item2'], 'tag2':'a normal string'}

有没有一种简单的方法可以做到这一点？
例如，以原始格式保存到字典的命令

或者我是否需要设置 if 语句来确定值类型并使用 Element.text 的替代方案单独保存它？

Answer 1

您可以使用literal_eval 尝试解析复杂的Python 文字。由于你的字符串没有被引用，它们会在 lteral eval 中引发 SyntaxError，但这很容易解决：

import xml.etree.cElementTree as xml
from ast import literal_eval

xmldata = {}
xmlfile = xml.parse(XMLFile.xml)
for xmltag in xmlfile.iter():
    try:
        xmldata[xmltag.tag] = literal_eval(xmltag.text)
    except SyntaxError:
        xmldata[xmltag.tag] = xmltag.text

与 Python 的内置“eval”不同，ast.literal_eval 不允许执行表达式，因此是安全的，即使 XML数据来自不受信任的来源。

Answer 2

这里是一个建议的解决方案：检查 [ 是否存在，然后解析列表。它不是万无一失的（如果分隔符不完全是带有空格的 , ，它将无法工作），但我认为您可以很容易地改进它。

import xml.etree.cElementTree as xml

xmldata = {}
xmlfile = xml.parse("data.xml")
for xmltag in xmlfile.iter():
    # it's a list
    if "[" in xmltag.text:
        d = xmltag.text.lstrip("[").rstrip("]")
        l = [item.lstrip("'").rstrip("'") for item in d.split(", ")]
        xmldata[xmltag.tag] = l
    else:
        xmldata[xmltag.tag] = xmltag.text

print xmldata

打印：{'root': '\n', 'tag1': ['item1', 'item2'], 'tag2': '普通字符串'}

Answer 3

我认为您没有充分利用 xml 的强大功能！

为什么不这样组织你的 .xml ：

<root>
    <tag1>
        <item>item1</item>
        <item>item2</item>
    </tag1>
    <tag2>a normal string<tag2>
</root>

这样你的 python 代码将把每个 作为 < 的容器来处理/code>，我认为这样更好。

注意：您可能还想查看此处。 （我同意作者的“最喜欢的方式”）