首頁 > 軟體

Python對XML檔案實現增刪改查操作

2022-11-11 14:00:37

PYTHON 操作 XML

讀取XML檔案

關於XML的介紹

<data> 與 </data> 是一對標籤的開始與結束

<property … /> 也是一個正確的標籤,以 /> 結尾,是在標籤沒有巢狀內容時的簡寫形式

name=“cat”,name是<data>標籤的一個屬性,cat是name屬性的值

description here …是<data>標籤的內容,這裡是一段文字。當然也可以是xml的巢狀

<data name="cat" num="10"> description here ... </data>

<property value="node" />

<country name="china">
	<province name="beijing">
		<school name="the sunshine school" />
	</province>
</country>

準備一個demo.xml檔案

<data>
    <teacher name="Albert">
        <birthday>1980</birthday>
        <gender>male</gender>
        <subject>Math</subject>
    </teacher>

    <student name="Becky">
        <birthday>2000</birthday>
        <gender>female</gender>
        <hobbies>
            <hobby>skating</hobby>
            <hobby>rocks</hobby>
        </hobbies>
        <exam absence="no">
            <math>90</math>
            <english>90</english>
            <music>95</music>
        </exam>

    </student>
    <student name="Cindy">
        <birthday>2001</birthday>
        <gender>female</gender>
        <hobbies>
            <hobby>reading</hobby>
            <hobby>guitar</hobby>
        </hobbies>
        <exam absence="yes">
        </exam>
    </student>

    <student name="Duke">
        <birthday>2000</birthday>
        <gender>male</gender>
        <hobbies>
            <hobby>football</hobby>
            <hobby>surfing</hobby>
        </hobbies>
        <exam absence="no">
            <math>100</math>
            <english>80</english>
            <music>92</music>
        </exam>
    </student>

</data>

讀取xml檔案內容

# Read the .xml file
tree = ET.parse("demo.xml")
root = tree.getroot()
print(root)

結果

<Element 'data' at 0x102d80cf8>

遍歷XML元素

for … in … 可以遍歷當前元素的所有直接子節點

for n in root:
    # items() returns all <key, value> pairs of the tag
    print(n, n.tag , n.attrib, n.items())

結果

(<Element 'teacher' at 0x1048b9e48>, 'teacher', {'name': 'Albert'}, [('name', 'Albert')])
(<Element 'student' at 0x1048bf0f0>, 'student', {'name': 'Becky'}, [('name', 'Becky')])
(<Element 'student' at 0x1048bf3c8>, 'student', {'name': 'Cindy'}, [('name', 'Cindy')])
(<Element 'student' at 0x1048bf5f8>, 'student', {'name': 'Duke'}, [('name', 'Duke')])

想要迭代遍歷當前元素的所有子節點(包括子孫節點)

for n in root.iter():
    print(n, n.tag)

結果

(<Element 'data' at 0x1052f0cf8>, 'data')
(<Element 'teacher' at 0x1052f0e48>, 'teacher')
(<Element 'birthday' at 0x1052f0d30>, 'birthday')
(<Element 'gender' at 0x1052f6080>, 'gender')
(<Element 'subject' at 0x1052f60b8>, 'subject')
(<Element 'student' at 0x1052f60f0>, 'student')
(<Element 'birthday' at 0x1052f6048>, 'birthday')
(<Element 'gender' at 0x1052f6128>, 'gender')
(<Element 'hobbies' at 0x1052f6198>, 'hobbies')
(<Element 'hobby' at 0x1052f6208>, 'hobby')
(<Element 'hobby' at 0x1052f6240>, 'hobby')
(<Element 'exam' at 0x1052f62b0>, 'exam')
(<Element 'math' at 0x1052f6320>, 'math')
(<Element 'english' at 0x1052f6390>, 'english')
(<Element 'music' at 0x1052f6400>, 'music')
(<Element 'student' at 0x1052f63c8>, 'student')
(<Element 'birthday' at 0x1052f6438>, 'birthday')
(<Element 'gender' at 0x1052f6470>, 'gender')
(<Element 'hobbies' at 0x1052f64a8>, 'hobbies')
(<Element 'hobby' at 0x1052f6518>, 'hobby')
(<Element 'hobby' at 0x1052f6588>, 'hobby')
(<Element 'exam' at 0x1052f65c0>, 'exam')
(<Element 'student' at 0x1052f65f8>, 'student')
(<Element 'birthday' at 0x1052f6630>, 'birthday')
(<Element 'gender' at 0x1052f6668>, 'gender')
(<Element 'hobbies' at 0x1052f66a0>, 'hobbies')
(<Element 'hobby' at 0x1052f6710>, 'hobby')
(<Element 'hobby' at 0x1052f6780>, 'hobby')
(<Element 'exam' at 0x1052f67b8>, 'exam')
(<Element 'math' at 0x1052f6828>, 'math')
(<Element 'english' at 0x1052f6898>, 'english')
(<Element 'music' at 0x1052f6908>, 'music')

想要選擇性地迭代直接子節點

for n in root.iter('teacher'):
    print(n, n.tag)
(<Element 'teacher' at 0x100f29e48>, 'teacher')

查詢XML元素

find與findall查詢xml元素

# find the first element
print(root.find('student'))
# find all  elements
print(root.findall('student'))
<Element 'student' at 0x1034300f0>

[<Element 'student' at 0x1034300f0>, <Element 'student' at 0x1034303c8>, <Element 'student' at 0x1034305f8>]

demo

for n in root:
    if n.tag == 'student' and n.get('name') == 'Becky':
        exam_node = n.find('exam')
        for subject in exam_node:
            print(subject.tag + " " + subject.text)

結果

math 90
english 90
music 95

新增XML元素

p = ET.Element(tag_name)

demo

for n in root:
    if n.tag == 'student' and n.get('name') == 'Cindy':
        exam_node = n.find('exam')
        exam_node.set("absence", "no")
        for subject in ['math', 'music']:
            p = ET.Element(subject)
            p.text = '90'
            exam_node.append(p)

if os.path.exists('new.xml'):
    os.remove('new.xml')
tree.write('new.xml', encoding='utf-8', xml_declaration=True)

結果

    <student name="Cindy">
        <birthday>2001</birthday>
        <gender>female</gender>
        <hobbies>
            <hobby>reading</hobby>
            <hobby>guitar</hobby>
        </hobbies>
        <exam absence="no">
        <math>90</math><music>90</music></exam>
    </student>

修改XML元素

demo

for n in root:
    if n.tag == 'student' and n.get('name') == 'Cindy':
        exam_node = n.find('exam')
        exam_node.set("absence", "no")
        exam_node.set("date", "2022-11-11")
        for subject in ['math', 'music']:
            p = ET.Element(subject)
            p.text = '90'
            exam_node.append(p)

        hobbies_node = n.find('hobbies').findall("hobby")
        hobbies_node[0].text = 'piano'
        p = ET.Element("hobby")
        p.set("old_hobby", 'yes')
        p.text = 'reading'
        n.find('hobbies').remove(hobbies_node[1])
        n.find('hobbies').append(p)

結果

    <student name="Cindy">
        <birthday>2001</birthday>
        <gender>female</gender>
        <hobbies>
            <hobby>piano</hobby>
            <hobby old_hobby="yes">reading</hobby></hobbies>
        <exam absence="no" date="2022-11-11">
        <math>90</math><music>90</music></exam>
    </student>

到此這篇關於Python對XML檔案實現增刪改查操作的文章就介紹到這了,更多相關Python XML增刪改查內容請搜尋it145.com以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援it145.com!


IT145.com E-mail:sddin#qq.com