将 Python XML ElementTree 输出写入 CSV

Question

TL;DR I'm now able to output the information I want in the CSV but I'm just repeating the last XML file's data over and over again.

This is the latest version of the script:

import csv
import glob
import xml.etree.ElementTree as ET
filenames = glob.glob("..\Lib\macros\*.xml")

for filename in filenames:

  with open(filename, 'r') as content:
    element = ET.parse(content)
    root = element.getroot()
    print(root.attrib, filename)
  e = element.findall('commands/MatrixSwitch/')
  for i in e:
    print (i.tag, i.text)


    with open('results.csv', 'w', newline='') as file:
        for filename in filenames:
            writer = csv.writer(file)
            writer.writerow([root.attrib, filename])
            for i in e:
                writer.writerow([i.tag, i.text])

Say I have 10 XML files, I'm getting the output related to XML "File 10" 10 times in the CSV, not anything for XML "File 1-9" ... sure its something simple?

=========================================================================

I've written a small script which ingests a folder of XML files, searches for a particular element and then recalls some of the data. This is then printed to the console and written to a CSV, except I'm having trouble formatting my CSV correctly.

This is where I've got so far:

import csv
import glob
import xml.etree.ElementTree as ET
filenames = glob.glob("..\Lib\macros\*.xml")

for filename in filenames:

  with open(filename, 'r') as content:
    element = ET.parse(content)
    root = element.getroot()
    print(root.attrib, filename)
  e = element.findall('commands/MatrixSwitch/')
  for i in e:
    print (i.tag, i.text)
  with open('results.csv', 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerow([root.attrib, filename])

I'm looking to capture the following data:

• XML Filename
• Macro Name
• Monitor ID
• Camera ID

I'm only interested in the and when a "Matrix Switch" is referred to in the XML. Sometimes there might only be one monitor ID and one camera ID, sometimes there might be more so the script needs to loop through and get all of the IDs within the "Matrix Switch" element. This seems to work so far.

Typical XML structure looks like this :

<macro name="NAME OF THE MACRO IS SHOWN HERE">
<execution>
<delay>0</delay>
</execution>
<parameters/>
<commands>
<MatrixSwitch>
<camera>1530</camera>
<monitor>1020</monitor>
</MatrixSwitch>
<MatrixSwitch>
<camera>1531</camera>
<monitor>1001</monitor>
</MatrixSwitch>
</commands>
</macro>

Or like this :

<macro name="ANOTHER NAME GOES HERE">
<execution>
<delay>0</delay>
</execution>
<parameters/>
<commands>
<MatrixSwitch>
<camera>201</camera>
<monitor>17</monitor>
</MatrixSwitch>
<MatrixSwitch>
<camera>206</camera>
<monitor>18</monitor>
</MatrixSwitch>
<MatrixSwitch>
<camera>202</camera>
<monitor>19</monitor>
</MatrixSwitch>
<MatrixSwitch>
<camera>207</camera>
<monitor>20</monitor>
</MatrixSwitch>
</commands>
</macro>

My current results.csv is only set to output the name and filename. This works but I'm unsure where I need to add the "writer" command to the loop where its dealing with the Monitor ID and Camera ID .

I want my CSV to show : Name, Filename, Monitor A, Camera A, Monitor B, Camera B, Monitor C, Camera C, Monitor D, Camera D etc.....

Any pointers greatly appreciated!!

Code has now been changed slightly :

import csv
import glob
import xml.etree.ElementTree as ET
filenames = glob.glob("..\Lib\macros\*.xml")

for filename in filenames:

  with open(filename, 'r') as content:
    element = ET.parse(content)
    root = element.getroot()
    print(root.attrib, filename)
  e = element.findall('commands/MatrixSwitch/')
  for i in e:
    print (i.tag, i.text)
    with open('results.csv', 'w', newline='') as file:
        writer = csv.writer(file)
        writer.writerow([root.attrib, filename])
        for i in e:
           writer.writerow([i.tag, i.text])

Output in the CSV is as below :

https://imgur.com/a/SrPrgjm

Answer 1

Just add a loop calling writerow:

...
with open('results.csv', 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerow([root.attrib, filename])
    for i in e:
        writer.writerow([i.tag, i.text])