class_instances_to_dataframe

介绍一下如何把多个class instances 的值存成一个dataframe.

defaultdict

先介绍一个defaultdict.这个是collections的一个骚东西.你建好了以后,每次新建一个Key的时候,value会自动建立一个自定义类型的初始化值.

python

import collections

例如 自定义value为list

python

list_dict = collections.defaultdict(list)

python

list_dict

defaultdict(list, {})

python

list_dict["a"].append(1)

python

list_dict["b"].append(2)

python

list_dict

defaultdict(list, {'a': [1], 'b': [2]})

够骚吧!

多个class instance 转换为defaultdict

python

class my_class(object):
    def __init__(self,a,b,c):
        self.a = a
        self.b = b
        self.c = c

python

class1 = my_class(1,2,3)

python

class2 = my_class(2,3,4)

python

class3 = my_class(3,4,5)

首先把这些classes全部添加到list

python

class_list = [class1, class2, class3]

创建一个defaultdict

python

class_dict = collections.defaultdict(list)

python

for class_instance in class_list:
    for key, value in class_instance.__dict__.items():
        class_dict[key].append(value)

python

class_dict

defaultdict(list, {'a': [1, 1, 3], 'b': [2, 2, 4], 'c': [3, 3, 5]})

创建dataframe

python

import pandas as pd

python

frame = pd.DataFrame.from_dict(class_dict)

python

frame

	a	b	c
0	1	2	3
1	1	2	3
2	3	4	5

大功告成!!骚吧!!