从对象列表创建多索引数据帧

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数据帧。 我总是在数据帧中使用嵌套数组。。。如何到达这样的多索引数据帧：

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

Answer 1

您可以使用__dict__方法像这样拉出object的属性，然后设置multi-index

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