继承pandas.DataFrame实现按列设置小数显示精度的方法咨询

核心实现思路 #

• 自定义继承pandas.DataFrame的子类，通过_metadata属性注册自定义的numberformat字典属性，避免pandas常规操作（切片、拷贝等）清除该自定义属性
• 重写父类的to_string()方法，渲染文本输出前遍历numberformat中配置的列，对列值临时应用指定的格式化规则，再调用父类原生方法输出格式化结果
• 若需要适配Jupyter Notebook的HTML显示效果，可额外重写_repr_html_()方法，用相同逻辑渲染HTML表格内容

完整实现代码 #

import pandas as pd

class FormatDataFrame(pd.DataFrame):
    # 注册自定义属性，pandas操作时会自动保留该属性
    _metadata = ['numberformat']

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        # 初始化列格式配置字典
        self.numberformat = dict()

    def to_string(self, *args, **kwargs):
        # 生成临时副本，不会修改原数据的实际存储值
        temp_df = self.copy()
        for col, fmt in self.numberformat.items():
            if col in temp_df.columns:
                temp_df[col] = temp_df[col].apply(lambda x: f"{x:{fmt}}")
        # 调用父类原生方法输出格式化内容
        return super(FormatDataFrame, temp_df).to_string(*args, **kwargs)

    # 可选方法：适配Jupyter环境的HTML显示
    def _repr_html_(self):
        temp_df = self.copy()
        for col, fmt in self.numberformat.items():
            if col in temp_df.columns:
                temp_df[col] = temp_df[col].apply(lambda x: f"{x:{fmt}}")
        return super(FormatDataFrame, temp_df)._repr_html_()

使用示例 #

# 构造测试数据
df = FormatDataFrame({
    'A': [1, 2, 3],
    'B': [2, 3, 4]
})

# 按需求语法指定对应列的显示格式
df.numberformat['A'] = '0.2f'

# 打印输出结果
print(df)

输出效果：

A  B
0  1.00  2
1  2.00  3
2  3.00  4

注意事项 #

• 切片、拷贝等操作返回的对象仍为FormatDataFrame类型，numberformat的配置会自动继承
• 格式化仅作用于显示环节，不会修改原DataFrame存储的实际数值
• 可根据需求自行扩展格式化逻辑，适配非数值类型列的显示需求

内容的提问来源于stack exchange，提问作者Anant Kumar