Python将对象集合数据写入到csv文件

悠悠 2023-09-29 10:49 100阅读 0赞

在我们实际业务开发中，肯定会遇到将集合对象的数据遍历完后写入到数据库或者导出为一个csv文件里，如下是一个具体实现的示例demo

### 代码示例 ###

import pandas as pd
    
    
    class Test:
        calorie = 0
        duration = 0
        flag = 0
    
    def init_object():
        """
         init class atrributes
        :return: list
        """
        test = Test()
        test.calorie = 20
        test.duration = 50
        test.flag = 1
        list = []
        # add obj to list
        list.append(test.__dict__)
        test = Test()
        test.calorie = 30
        test.duration = 60
        test.flag = 0
        # add obj to list
        list.append(test.__dict__)
        print(list)
    
        return list
    
    def obj_csv( obj_list):
        """
    
        :param obj_list:
        :return: csv
        """
        # convert list to Dataframe
        df = pd.DataFrame(obj_list)
        print(df)
        # take dataframe to csv file
        df.to_csv("output.csv", index=None)
    
    obj_list = init_object()
    obj_csv(obj_list)

### 运行结果 ###

![在这里插入图片描述][watermark_type_d3F5LXplbmhlaQ_shadow_50_text_Q1NETiBAUGF0cmljayBKaWEgfn4_size_20_color_FFFFFF_t_70_g_se_x_16]

![在这里插入图片描述][watermark_type_d3F5LXplbmhlaQ_shadow_50_text_Q1NETiBAUGF0cmljayBKaWEgfn4_size_19_color_FFFFFF_t_70_g_se_x_16]

[watermark_type_d3F5LXplbmhlaQ_shadow_50_text_Q1NETiBAUGF0cmljayBKaWEgfn4_size_20_color_FFFFFF_t_70_g_se_x_16]: https://img-blog.csdnimg.cn/f369779155174131a0965919e34c43a1.png?x-oss-process=image/watermark,type_d3F5LXplbmhlaQ,shadow_50,text_Q1NETiBAUGF0cmljayBKaWEgfn4=,size_20,color_FFFFFF,t_70,g_se,x_16
[watermark_type_d3F5LXplbmhlaQ_shadow_50_text_Q1NETiBAUGF0cmljayBKaWEgfn4_size_19_color_FFFFFF_t_70_g_se_x_16]: https://img-blog.csdnimg.cn/afb6c10f2a3b464792622f62d886396a.png?x-oss-process=image/watermark,type_d3F5LXplbmhlaQ,shadow_50,text_Q1NETiBAUGF0cmljayBKaWEgfn4=,size_19,color_FFFFFF,t_70,g_se,x_16