Как разобрать несколько XML-файлов на несколько CSV-файлов?
Я проанализировал файл XML, используя этот код, который работает для одного ввода xml для одного вывода csv. Я пытался использовать glob для работы с несколькими входными данными, а также с несколькими выходными файлами csv, но я знаю, что это неправильно.
import glob
import xml.etree.ElementTree as et
import csv
for file in glob.glob('./*.xml'):
with open(file) as f:
tree = et.parse(f)
nodes = tree.getroot()
with open(f'{f[:-4]}edited.csv', 'w') as ff:
cols = ['dateTime','x','y','z','motion','isMoving','stepCount','groupAreaId','commit']
nodewriter = csv.writer(ff)
nodewriter.writerow(cols)
for node in nodes:
values = [ node.attrib.get(kk, '') for kk in cols]
nodewriter.writerow(values)
Как мне измениться, чтобы получить несколько выходных данных csv?
Получил ваш ответ!
Ответы 2
вы можете создать список имен файлов, а затем записать туда файлы xml. Если выходные файлы уже находятся в директории, то с помощью glob можно получить имена. Если файлы не существуют, приведенный ниже код будет создан с использованием данного имени файла.
csvFileNames = ['outputfile1.csv', 'outputfile2.csv']
for file in csvFileNames:
with open(file, 'w') as f:
wtr = csv.writer(f)
wtr.writerows( [[1, 2], [2, 3], [4, 5]]) # write what you want
чтобы получить имена файлов XML из каталога, вы можете попробовать следующий код:
from os import listdir
filenames = listdir('.') # here dot is used because script and csv files are in the same directory, if XML files are in other directory then set the path inside listdir
xmlFileNames = [ filename for filename in filenames if filename.endswith( ".xml" ) ]
# get xml file names like this, xmlFileNames = ["abc.xml", "ef.xml"]
resultCsvFileNameList = [fname.replace(".xml", ".csv") for fname in xmlFileNames ]
но если мне нужно написать тысячи CSV-файлов, что может быть проще, чем упоминать имена CSV-файлов по одному?
вам не нужно писать имя файла вручную, используя glob вы можете собирать имена файлов
спасибо за редактирование, но моя проблема в том, что мне нужно создать файл .csv из файла xml, и, очевидно, файл csv еще не существует. вот почему мне интересно, как автоматически создать файл .csv с тем же именем, что и xml.
Ваш код в настоящее время использует дескриптор файла для формирования имени выходного файла. Вместо f используйте file следующим образом:
import glob
import xml.etree.ElementTree as et
import csv
for file in glob.glob('./*.xml'):
with open(file) as f:
tree = et.parse(f)
nodes = tree.getroot()
with open(f'{file[:-4]}edited.csv', 'w') as ff:
cols = ['dateTime','x','y','z','motion','isMoving','stepCount','groupAreaId','commit']
nodewriter = csv.writer(ff)
nodewriter.writerow(cols)
for node in nodes:
values = [ node.attrib.get(kk, '') for kk in cols]
nodewriter.writerow(values)
Вы хотели использовать
with open(f'{file[:-4]}edited.csv', 'w') as ff:? В настоящее время вы используете дескриптор файла для имени файла.