запись данных в csv из словарей с несколькими значениями на ключ - PullRequest
Новая
       19

запись данных в csv из словарей с несколькими значениями на ключ

0 голосов
/ 26 октября 2018

Фон

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

Проблема / решение

Изображение 1 - это то, как распечатывается мой фактический вывод.На рисунке 2 показано, как я хочу, чтобы мой вывод был распечатан. Изображение 2 - желаемый вывод .

enter image description here enter image description here

CODE 

import csv
from itertools import izip_longest

e = {'Lebron':[25,10],'Ray':[40,15]}
c = {'Nba':5000}

def writeData():
    with open('file1.csv', mode='w') as csv_file:
        fieldnames = ['Player Name','Points','Assist','Company','Total Employes']
        writer = csv.writer(csv_file)
        writer.writerow(fieldnames)

        for employee, company in izip_longest(e.items(), c.items()):
            row = list(employee)
            row += list(company) if company is not None else ['', '']  # Write empty fields if no company

            writer.writerow(row)

writeData()

Я открыт для всех решений / предложений, которые могут помочь мне получить желаемый формат вывода.

Ответы [ 2 ]

0 голосов
/ 28 октября 2018

Для гораздо более простого ответа вам просто нужно добавить одну строку кода к тому, что у вас есть: 

row = [row[0]] + row[1]

так: 

for employee, company in izip_longest(e.items(), c.items()):
        row = list(employee)
        row = [row[0]] + row[1]
        row += list(company) if company is not None else ['', '']  # Write empty fields if no company
0 голосов
/ 26 октября 2018 
from collections import defaultdict

values = defaultdict(dict)
values[Name1] = {Points: [], Assist: [], Company: blah, Total_Employees: 123}

для генерации вывода, просмотрите каждый элемент в значениях, чтобы дать вам имена, и заполните другие значения, используя key_values ​​во вложенном диктовке.

Опять же, убедитесь, что нет нескольких записейс тем же именем или выберите тот, который содержит уникальные записи в defaultdict.

Демонстрация для примера- 

>>> from collections import defaultdict
>>> import csv
>>> values = defaultdict(dict)
>>> vals = [["Lebron", 25, 10, "Nba", 5000], ["Ray", 40, 15]]
>>> fields = ["Name", "Points", "Assist", "Company", "Total Employes"]
>>> for item in vals:
...     if len(item) == len(fields):
...             details = dict()
...             for j in range(1, len(fields)):
...                     details[fields[j]] = item[j]
...             values[item[0]] = details
...     elif len(item) < len(fields):
...             details = dict()
...             for j in range(1, len(fields)):
...                     if j+1 <= len(item):
...                             details[fields[j]] = item[j]
...                     else:
...                             details[fields[j]] = ""
...             values[item[0]] = details
... 
>>> values
defaultdict(<class 'dict'>, {'Lebron': {'Points': 25, 'Assist': 10, 'Company': 'Nba', 'Total Employes': 5000}, 'Ray': {'Points': 40, 'Assist': 15, 'Company': '', 'Total Employes': ''}})
>>> csv_file =  open('file1.csv', 'w')
>>> writer = csv.writer(csv_file)
>>> for i in values:
...     row = [i]
...     for j in values[i]:
...             row.append(values[i][j])
...     writer.writerow(row)
... 
23
13
>>> csv_file.close()

Содержимое файла file1.csv: 

Lebron,25,10,Nba,5000
Ray,40,15,,
...