如何让dataclass实例动态新增属性在DataFrame中正常显示？

问题背景 #

使用Python的dataclass定义Business类后，通过setattr给实例动态添加了Company属性，但将包含该实例的列表转换为Pandas DataFrame时，仅显示Job_Trade列，新增的Company列无法被识别。

核心原因 #

Pandas转换dataclass实例为DataFrame时，默认只会识别dataclass定义阶段声明的字段，通过setattr动态添加的属性不在预定义字段列表中，因此不会被自动纳入DataFrame的列。

解决方法 #

方法1：使用vars()转换实例为字典 #

vars()函数会返回对象的所有属性（包括动态添加的）组成的字典，将每个Business实例先转成字典再生成DataFrame即可：

修改生成DataFrame的代码行，完整代码如下：

import dataclasses
from dataclasses import dataclass, field
import pandas as pd

@dataclass
class Business(object):
    Job_Trade: str = None

    def __repr__(self):
        attrs = [f'{attr}={getattr(self, attr)!r}' for attr in dir(self) if not callable(getattr(self, attr)) and not attr.startswith("__")]
        return f'{self.__class__.__name__}({", ".join(attrs)})'

def add_variable(instance, variable_name, value):
     setattr(instance, variable_name, value)

@dataclass
class BusinessList:
    business_list : list[Business] = field(default_factory=list)

business = Business()
business.Job_Trade = 'Licensed Customer Service Representative'
add_variable(business, 'Company', 'Global Payments')

business_list = BusinessList()
business_list.business_list.append(business)

# 关键修改：用vars()转换每个实例为字典
df = pd.DataFrame([vars(b) for b in business_list.business_list])
print(df)

运行后输出：

Job_Trade          Company
0  Licensed Customer Service Representative  Global Payments

方法2：自定义字典转换方法 #

如果需要更灵活的属性筛选控制，可以在Business类中添加自定义方法返回所有属性：

@dataclass
class Business(object):
    Job_Trade: str = None

    def __repr__(self):
        attrs = [f'{attr}={getattr(self, attr)!r}' for attr in dir(self) if not callable(getattr(self, attr)) and not attr.startswith("__")]
        return f'{self.__class__.__name__}({", ".join(attrs)})'
    
    def to_dict(self):
        # 过滤掉以下划线开头的内部属性
        return {k: v for k, v in self.__dict__.items() if not k.startswith("_")}

生成DataFrame时调用该方法：

df = pd.DataFrame([b.to_dict() for b in business_list.business_list])

补充说明 #

• 原生dataclasses.asdict()仅返回dataclass预定义字段，无法获取动态添加的属性，不适用此场景。
• vars()和__dict__都能获取实例的全部属性，是解决动态属性不显示问题的核心。

内容的提问来源于stack exchange，提问作者Ali Anwar