python - 从对象列表创建多索引 pandas 数据框

Question

这里是获取对象列表的示例代码:

import numpy as np

class simpleobj():

    def __init__(self,name):
        self.attr1 = name
        self.attr2 = f"{name}_{np.random.randint(1,100)}"
        self.x= np.random.randint(1,100,size=(100,1))
        self.y1= np.random.randint(1,100,size=(100,1))
        self.y2= np.random.randint(1,100,size=(100,1))

objectlist=[simpleobj(i) for i in list('SAMPLETHINGTOWIRTENOIDEA') ]

我想将这些对象列表转换为 mi-dataframe。我总是在数据框中得到嵌套数组...如何达到这样的多索引数据框:

|----索引----|
属性1 |属性2 | x| y1 | y2
'S'| 'S50' | 1 | 4 | 3
'S'| 'S50' | 2 | 5 |10

Answer 1

您可以像这样使用__dict__方法提取object的属性，然后设置多索引。

import pandas as pd
df = pd.DataFrame([obj.__dict__ for obj in objectlist]).set_index(['attr1', 'attr2'])