从每个键有多个值的字典中将数据写入 csv

Question

背景

我正在将数据存储在字典中。字典可以有不同的长度，并且在特定字典中可以有具有多个值的键。我正在尝试将数据吐出到 CSV 文件中。

Problem/Solution

图 1 是我的实际输出结果。图 2 显示了我希望我的输出如何实际打印输出。 图像 2 是所需的输出。

代码

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()

我对所有可以帮助我获得所需输出格式的 solutions/suggestions 开放。

Answer 1

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,,

Answer 2

要获得更简单的答案，您只需在现有内容中添加一行代码：

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

从每个键有多个值的字典中将数据写入 csv

writing data to csv from dictionaries with multiple values per key

python

csv

import-csv

izip