python将字典写入csv文件

在数据处理完成后,常常需要将结果储存成字典的形式并保存下来,一个简单的办法是,将字典转换成pandas的DataFrame,然后直接使用to_csv方法即可,但这样需要使用pandas第三方库,如果不使用第三方库呢?

代码实现:

import csv


def dict2csv(dic, filename):
    """
    将字典写入csv文件,要求字典的值长度一致。
    :param dic: the dict to csv
    :param filename: the name of the csv file
    :return: None
    """
    file = open(filename, 'w', encoding='utf-8', newline='')
    csv_writer = csv.DictWriter(file, fieldnames=list(dic.keys()))
    csv_writer.writeheader()
    for i in range(len(dic[list(dic.keys())[0]])):   # 将字典逐行写入csv
        dic1 = {key: dic[key][i] for key in dic.keys()}
        csv_writer.writerow(dic1)
    file.close()


my_dict = {'a': [1, 2, 4, 5, 5, 4, 1],
          'b': [2, 4, 5, 6, 7, 7, 1],
          'c': [4, 4, 1, 6, 7, 1, 6]}
dict2csv(my_dict,'D:/pythonjpgs/dict2csvfunctest1.csv')


显然,我们使用python,将字典成功写入csv文件。

by——神采的二舅

你可能感兴趣的:(python,pandas,开发语言)