Извлечение XML в словарь - PullRequest
Новая
       11

Извлечение XML в словарь

0 голосов
/ 15 января 2019

Я написал код, и некоторые значения в ecd отсутствуют. Я хотел бы указать их как 'None' или 0000, чтобы иметь возможность создавать кадры данных. К сожалению, код работает до тех пор, пока отсутствует место, а затем происходит сбой, и я не могу обнаружить ошибку.

Сообщение об ошибке: 

File "extra.py", line 236, in <module>
    if dic['mudLogs']['mudLog']['geologyInterval'][i]['ecdTdAv']['#text'] != None:
  KeyError: 'ecdTdAv'

Код: 

xml_file = 'C:\\Users\\jtfra\\Desktop\\Thesis\\Volve_Real_Time_DData\\WITSML Realtime drilling data\\Norway-Statoil-NO 15_$47$_9-F-11\\1\\mudLog\\1.xml'

def convert(xml_file, xml_attribs=True):
with open(xml_file, "rb") as f:    # notice the "rb" mode
    d = xmltodict.parse(f, xml_attribs=xml_attribs)
    return d

dic = convert(xml_file)
mdTop, ecd = [], []


 for i in range(len(dic['mudLogs']['mudLog']['geologyInterval'])):
    mdTop.append(dic['mudLogs']['mudLog']['geologyInterval'][i]['mdTop']['#text'])

    if dic['mudLogs']['mudLog']['geologyInterval'][i]['ecdTdAv']['#text'] != None:
        ecd.append(dic['mudLogs']['mudLog']['geologyInterval'][i]['ecdTdAv']['#text'])
    else:
        ecd.append('None')

print(ecd)

1 Ответ

0 голосов
/ 15 января 2019

Вместо доступа к нему как: 

dic['mudLogs']['mudLog']['geologyInterval'][0]['ecdTdAv']

сделать: 

dic['mudLogs']['mudLog']['geologyInterval'][0].get('ecdTdAv', '0000')

или аналогичный.

Вы также можете проверить наличие ключа: 

if 'ecdTdAv' in dic['mudLogs']['mudLog']['geologyInterval'][I]:
    # do something with it, e.g.:
    print(dic['mudLogs']['mudLog']['geologyInterval'][i]['ecdTdAv']['#text'])
...