Python - 猴子补丁失败了，为什么？

Question

我想在已安装的模块上对f(*args, **kwargs)进行修补。我在自己的代码上使用了decorator的想法，但是已安装模块的其他方法无法正确调用f。

这是一个例子：

import numpy as np

def log(func):
    def wrapper(*args, **kwargs):
        print('logging')
        return func(*args, **kwargs)
    return wrapper

if __name__ == "__main__":
    a1 = np.asarray([0, 1, 2])
    print(f'a1={a1}')

    a2 = np.array([0, 1, 2])
    print(f'a2={a2}')

    np.array = log(np.array)

    a3 = np.asarray([0, 1, 2])
    print(f'a3={a3}')

    a4 = np.array([0, 1, 2])
    print(f'a4={a4}')

输出是：

a1=[0 1 2]
a2=[0 1 2]
a3=[0 1 2]
logging
a4=[0 1 2]

我希望结果如下：

a1=[0 1 2]
a2=[0 1 2]
logging
a3=[0 1 2]
logging
a4=[0 1 2]

因为asarray在源代码中调用array。

我的问题是：1。为什么猴子补丁失败？ 2.如何解决？

Answer 1

通过np.array = log(np.array)你正在改变“公共”名称np.array所指的功能。

但np.asarray在numpy.core.numeric模块中定义，该模块有自己的“私有”名称来引用该函数。它不受修补公共名称的影响。

您必须修补私人名称：

np.core.numeric.array = log(np.array)