python - 用csv写两个字典

Question

您好，我需要将两个词典写入一个 csv，我想按顺序编写第一个词典，而第二个词典的顺序无关紧要，第二个字典的键是来自 selenium 驱动程序的 cookie，因为我不知道每个站点将返回多少个键(每个网站都不同)

user1= {'ID':1,'Name':'John','Age':13} # the order I want to keep those must come first
cookies_data = driver.get_cookies() # dictionary
with open('file.csv','w') as csvFile:
    #write to file in this order id,name,age,cookies


# A sample of cookies:
#cookies_data = [{'domain': 'google.com', 'expiry': 1624301720.007404, 'httpOnly': True, 'name': 'ANID', 'path': '/',}] 
#cookies_data can have more than one dictionary and some of the keys may be the same 

Answer 1

您可以使用 csv.DictWriter编写您的 csv，从组合字典中选择列的名称和值，并使用它们编写 csv，同时遍历 cookies_data 列表

#Combine fieldnames of both dictionaries
fieldnames = list(user1.keys()) + list(cookies_data[0].keys())

#Open csv file
with open('file.csv','w') as csvFile:

    #Write the header and write the combined dictionary
    writer = csv.DictWriter(csvFile, fieldnames=fieldnames)
    writer.writeheader()

    #Loop over the cookies_data list and write the combined dict to csv
    for data in cookies_data:
        new_dict = {**user1, **data}
        writer.writerow(new_dict)

输出将是

ID,Name,Age,domain,expiry,httpOnly,name,path,secure,value
1,John,13,.....
1,John,13,.....
1,John,13,.....
1,John,13,.....