Question

我试图通过将密钥与csv标头中的列匹配来将多个字典（键和值）推送到csv文件。例如：

import csv
d1 = {'a':1, 'b':2, 'c': 3}
d2 = {'d':4, 'e':5, 'f': 6}
with open('my_data.csv','wb') as f:
    w = csv.writer(f)
    w.writerow(['a', 'b', 'c', 'd', 'e', 'f'])

#iterate through all keys in d1,d2,dn
#if key matches column:
    #write value of key to the bottom of column
#else:
    #error key not found in header

mydata.csv中的预期结果

a,b,c,d,e,f
1,2,3,4,5,6

Answer 1

答案是..不要只是将列名传递给writerow（）..将它们放在变量columns中，然后使用它来控制写出值的顺序。 Python词典没有顺序..您必须使用一些代码将值排序为您想要的顺序。

将值写入CSV的最后一行代码使用名为List Comprehension的python功能。这是一个节省3-4行代码的快捷方式。查一查，它们非常方便。

import csv
d1 = {'a':1, 'b':2, 'c': 3}
d2 = {'d':4, 'e':5, 'f': 6}

columns = ['a', 'b', 'c', 'd', 'e', 'f']

# combine d1 and d2 into data.. there are other ways but this is easy to understand
data = dict(d1)
data.update(d2)

with open('my_data.csv','wb') as f:
    w = csv.writer(f)
    w.writerow(columns)
    # make a list of values in the order of columns and write them
    w.writerow([data.get(col, None) for col in columns])

如果没有列表理解，这就是它的样子：

    row = []
    for col in columns:
        row.append(data.get(col, None)) 
    w.writerow(row)

具有列匹配的csv文件的Python字典键

1 个答案: