如何替换第三方库对象的in_range方法？遇覆盖失败问题求助

问题描述 #

想要替换第三方库中power_spectrum对象的in_range方法，实现将频率范围的最小值和最大值各扩展3的功能。原方法代码如下：

def in_range(self, frequency_min, frequency_max):
    """Returns part of the power spectrum within a given frequency range."""
    ir = copy(self)
    mask = (self.frequency > frequency_min) & (self.frequency <= frequency_max)
    ir.frequency = self.frequency[mask]
    ir.power = self.power[mask]
    return ir

自行编写的替换代码运行时出现错误plot_power_spectrum..in_range() missing 1 required positional argument: 'frequency_max'，且通过print(inspect.getsource(power_spectrum.in_range))检查发现新方法并未覆盖原方法，不符合需求。替换代码如下：

def in_range(self, frequency_min, frequency_max):
    
    ir = list.copy(self)
    mask = (self.frequency > frequency_min - 3) & (self.frequency <= frequency_max + 3)
    ir.frequency = self.frequency[mask]
    ir.power = self.power[mask]
    return ir

power_spectrum.in_range = in_range

错误原因分析 #

1. 拷贝方式错误：原方法使用copy(self)（即copy模块的浅拷贝），但替换代码中误用了list.copy(self)，而self是power_spectrum类的实例，并非列表，会导致拷贝逻辑异常。
2. 方法绑定问题：若power_spectrum是实例对象而非类，直接将普通函数赋值给实例属性时，Python不会自动绑定self参数，调用时就会出现参数缺失错误；若操作的是类，但函数定义或赋值逻辑存在问题，也会导致覆盖失败。

正确解决方法 #

情况1：power_spectrum是类对象 #

直接修改类的方法属性，确保拷贝逻辑与原方法一致：

import copy
# 导入第三方库中的power_spectrum类
from your_third_party_lib import power_spectrum

def in_range(self, frequency_min, frequency_max):
    """扩展频率范围±3后的in_range方法"""
    ir = copy.copy(self)  # 沿用原方法的浅拷贝逻辑
    mask = (self.frequency > frequency_min - 3) & (self.frequency <= frequency_max + 3)
    ir.frequency = self.frequency[mask]
    ir.power = self.power[mask]
    return ir

# 直接替换类的in_range方法
power_spectrum.in_range = in_range

情况2：power_spectrum是实例对象 #

需要使用types.MethodType将函数绑定到实例，确保self参数正确传递：

import copy
import types
# 获取power_spectrum实例
ps = your_function_to_get_power_spectrum_instance()

def in_range(self, frequency_min, frequency_max):
    """扩展频率范围±3后的in_range方法"""
    ir = copy.copy(self)
    mask = (self.frequency > frequency_min - 3) & (self.frequency <= frequency_max + 3)
    ir.frequency = self.frequency[mask]
    ir.power = self.power[mask]
    return ir

# 将函数绑定为实例的方法
ps.in_range = types.MethodType(in_range, ps)

验证覆盖结果 #

替换完成后，使用以下代码检查是否成功覆盖：

import inspect
# 若替换的是类
print(inspect.getsource(power_spectrum.in_range))
# 若替换的是实例
print(inspect.getsource(ps.in_range))

输出内容应为新编写的方法代码，说明覆盖成功。

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