python - 解析 xml 并编写带有标题列的 csv

Question

我正在尝试解析包含气象数据的 xml 文件并在 csv 文件中写入一些值。 我不确定这段代码是否优雅，但它确实有效。

from qgis.PyQt.QtCore import *
import requests
import xml.etree.ElementTree as ET
# url of xml to parse
baseUrl = ('http://www.arpa.veneto.it/bollettini/meteo/h24/img08/0144.xml')
resp = requests.get(baseUrl)
msg = resp.content
tree = ET.fromstring(msg)
for stazione in tree.iter('STAZIONE'):
    idstaz= stazione.find('IDSTAZ').text
    for sensore in stazione.iter('SENSORE'):
      id= sensore.find('ID').text  
      for dati in sensore.iter('DATI'):
        ist = dati.get('ISTANTE') 
        vm = dati.find('VM').text
        f = open('D:/GIS/_Temp/result.csv', 'a')
        print >> f, idstaz, id, ist, vm
        f.close()

我不确定这段代码是否优雅，但它确实有效。

144 300000864 201701080100 -4.2
144 300000864 201701080200 -4.5
144 300000864 201701080300 -4.8
144 300000864 201701080400 -5.5
...

但我不知道如何将标题添加到列中。

Answer 1

在for循环之前打开文件并添加文件头

from qgis.PyQt.QtCore import *
import requests
import xml.etree.ElementTree as ET
# url of xml to parse
baseUrl = ('http://www.arpa.veneto.it/bollettini/meteo/h24/img08/0144.xml')
resp = requests.get(baseUrl)
msg = resp.content
tree = ET.fromstring(msg)
f = open('D:/GIS/_Temp/result.cvs', 'a')
f.write('STAZIONE,IDSTAZ,SENSORE,ISTANTE')
for stazione in tree.iter('STAZIONE'):
    idstaz= stazione.find('IDSTAZ').text
    for sensore in stazione.iter('SENSORE'):
      id= sensore.find('ID').text  
      for dati in sensore.iter('DATI'):
        ist = dati.get('ISTANTE') 
        vm = dati.find('VM').text

        print >> f, idstaz, id, ist, vm

f.close()