Python新手求助：如何重载Matplotlib的plot函数支持自定义测量量类

嘿，先给你明确你做的这个功能的专业称谓：这属于带不确定度的数值表示与误差传播计算，在科学计算领域很常用，不少成熟库（比如uncertainties）就是干这个的，你自己动手实现这个类真的挺棒的！

接下来聊聊怎么让Matplotlib的plot函数支持你的magnitude类——Matplotlib默认只认数值类型（比如int、float、NumPy数组），所以我们只需要让自定义类能被它“理解”成数值就行，有两种简单的方法：

方法一：给magnitude类添加数值转换方法 #

直接在你的类里实现__float__和__array__方法，让Matplotlib/NumPy能自动把对象转换成数值：

from math import sqrt, log10, fabs, floor, ceil, pi

class magnitude:
    def __init__(self, val, err):
        self.val = val
        self.err = err
    def __str__(self):
        return f"{self.val} ± {self.err}"  # 补全字符串输出会更友好
    def __float__(self):
        # 返回测量值的数值部分，供绘图使用
        return float(self.val)
    def __array__(self):
        # 支持NumPy数组转换，批量处理更方便
        import numpy as np
        return np.array(self.val)

这样你直接把magnitude对象的列表传给plt.plot()就可以了：

import matplotlib.pyplot as plt

# 示例数据
data = [magnitude(1.2, 0.1), magnitude(2.5, 0.2), magnitude(3.3, 0.15)]
plt.plot(data, '-o')
plt.show()

方法二：给Matplotlib注册类型转换器 #

如果你不想修改类的定义，可以给Matplotlib注册一个专门处理magnitude类型的转换器，让它自动转换：

from matplotlib.cbook import register

# 定义转换函数：把magnitude对象转成数值
def convert_magnitude(obj):
    return float(obj.val)

# 注册转换器
register(magnitude, convert_magnitude)

之后同样可以直接用plt.plot()传入magnitude对象列表，Matplotlib会自动调用转换函数处理。

额外提示：绘制误差棒 #

既然你的类包含误差，其实更适合用Matplotlib的errorbar函数来同时展示数值和误差，你可以给类加一个提取误差的方法：

class magnitude:
    # 其他方法不变
    def get_error(self):
        return self.err

然后这样使用：

x = range(len(data))
y_vals = [float(d) for d in data]
y_errs = [d.get_error() for d in data]
plt.errorbar(x, y_vals, yerr=y_errs, fmt='-o', capsize=5)
plt.show()

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