Создать CSV из нестандартного списка словарей в python - PullRequest
0 голосов
/ 06 августа 2020

У меня есть список словарей, например

[
  {'AB Code': 'Test AB Code', 'Created': '2020-08-04 13:20:55.196500+00:00'},
  {'AB Code': 'Test AB Code', 'Created': '2020-08-04 13:20:11.315136+00:00', 'Name': 'John Doe', 'Email': 'john@example.com', 'Phone No.': '1234567890', 'Age': '31'}
]

Два объекта словаря имеют нестандартные ключи. Я хочу создать заголовок для каждого нового ключа и значений под ним.

Итоговый CSV должен быть

AB Code, Created, Name, Email, Phone No., Age
Test AB Code, 2020-08-04 13:20:55.196500+00:00, '', '', '', ''
Test AB Code, 2020-08-04 13:20:55.196500+00:00, John Doe, john@example.com, 1234567890, 31

Я делаю

# header
d_ = []

# values
for index, item in enumerate(data):
  if index == 0:
    d_.append(list(item.keys()))
  d_.append(list(item.values()))

# Add to CSV
buffer = io.StringIO()
wr = csv.writer(buffer, quoting=csv.QUOTE_ALL)
wr.writerows(d_)

Что генерирует CSV

AB Code, Created
Test AB Code, 2020-08-04 13:20:55.196500+00:00, '', '', '', ''
Test AB Code, 2020-08-04 13:20:55.196500+00:00, John Doe, john@example.com, 1234567890, 31

1 Ответ

1 голос
/ 06 августа 2020

В комментариях к вопросу есть ответ с использованием pandas, предоставленный @bigbounty. Вот решение, использующее только стандартную библиотеку

import csv
from collections import ChainMap

data = [
  {'AB Code': 'Test AB Code', 'Created': '2020-08-04 13:20:55.196500+00:00'},
  {'AB Code': 'Test AB Code', 'Created': '2020-08-04 13:20:11.315136+00:00', 'Name': 'John Doe', 'Email': 'john@example.com', 'Phone No.': '1234567890', 'Age': '31'}
]

keys = list(ChainMap(*data))
with open('spam.csv', 'w', newline='') as f:
    wrtr = csv.DictWriter(f, fieldnames=keys, quoting=csv.QUOTE_ALL)
    wrtr.writeheader()
    wrtr.writerows(data)

Также есть расширенное обсуждение слияния dicts , которое может быть несколько актуальным.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...