如何对含自定义Tag对象的Pandas DataFrame计算均值并返回Tag对象

嘿，我来帮你搞定这个计算DataFrame中Tag对象均值的需求！你已经实现了Tag类的__add__方法，接下来只需要简单几步就能得到均值对应的新Tag对象，下面分两种方案来实现：

方案一：直接提取value计算（无需修改Tag类） #

这种方式不需要给Tag类加新方法，直接遍历提取每个Tag的value属性计算均值，再生成新的Tag对象：

import pandas as pd

class Tag(object):
    def __init__(self, value):
        self.value = value
    def __add__(self, other):
        return Tag(self.value + other.value)

# 测试数据
a = Tag(2)
b = Tag(8)
frame = pd.DataFrame({'tag': [a, b]})

# 计算均值Tag
total_value = sum(tag.value for tag in frame['tag'])
mean_value = total_value / len(frame['tag'])
mean_tag = Tag(mean_value)

print(mean_tag.value)  # 输出: 5.0

方案二：利用加法扩展除法（更贴合Pandas风格） #

如果你想更贴合Pandas的操作习惯，可以给Tag类添加__truediv__方法，这样就能先通过sum()求和，再直接除以数量得到均值：

import pandas as pd

class Tag(object):
    def __init__(self, value):
        self.value = value
    def __add__(self, other):
        return Tag(self.value + other.value)
    # 添加除法方法，支持和数值类型的除法
    def __truediv__(self, other):
        if isinstance(other, (int, float)):
            return Tag(self.value / other)
        raise TypeError("仅支持与整数/浮点数进行除法运算")

# 测试数据
a = Tag(2)
b = Tag(8)
frame = pd.DataFrame({'tag': [a, b]})

# 计算均值Tag
sum_tag = frame['tag'].sum()  # 调用__add__得到总和Tag(value=10)
mean_tag = sum_tag / len(frame['tag'])

print(mean_tag.value)  # 输出: 5.0

为什么这样做？ #

Pandas内置的mean()方法默认只支持数值类型的数据，无法直接识别自定义的Tag对象。所以我们要么手动提取核心的value属性计算，要么通过扩展Tag类的运算符方法，让Pandas的聚合操作（比如sum()）能和后续的除法操作无缝配合。

内容的提问来源于stack exchange，提问作者Yassine Faris