Как умножить функции в Python?

def sub3(n):
    return n - 3

def square(n):
    return n * n

В python легко создавать функции:

>>> my_list
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
>>> [square(sub3(n)) for n in my_list]
[9, 4, 1, 0, 1, 4, 9, 16, 25, 36]

К сожалению, при желании использовать композицию в качествеключХамовато

>>> sorted(my_list, key=lambda n: square(sub3(n)))
[3, 2, 4, 1, 5, 0, 6, 7, 8, 9]

Это должно быть простоsorted(my_list, key=square*sub3)потому что, черт возьми, функция__mul__ в любом случае не используется ни для чего другого:

>>> square * sub3
TypeError: unsupported operand type(s) for *: 'function' and 'function'

Что ж, тогда давайте просто определимся!

>>> type(sub3).__mul__ = 'something'
TypeError: can't set attributes of built-in/extension type 'function'

D'о!

>>> class CoolerFunction(types.FunctionType):
...     pass
...
TypeError: Error when calling the metaclass bases
    type 'function' is not an acceptable base type

D'о!

class Hack(object):
    def __init__(self, function):
        self.function = function
    def __call__(self, *args, **kwargs):
        return self.function(*args, **kwargs)
    def __mul__(self, other):
        def hack(*args, **kwargs):
            return self.function(other(*args, **kwargs))
        return Hack(hack)

Эй, теперь мы где-то ...

>>> square = Hack(square)
>>> sub3 = Hack(sub3)
>>> [square(sub3(n)) for n in my_list]
[9, 4, 1, 0, 1, 4, 9, 16, 25, 36]
>>> [(square*sub3)(n) for n in my_list]
[9, 4, 1, 0, 1, 4, 9, 16, 25, 36]
>>> sorted(my_list, key=square*sub3)
[3, 2, 4, 1, 5, 0, 6, 7, 8, 9]

Но я не хочуHack вызываемый класс! Правила области видимости совершенно разные, и я не до конца их понимаю, и это даже страшнее, чем, возможно, «ламеда». Я хочу, чтобы обезьянафункции, Как я могу это сделать?

Ответы на вопрос(3)

Ваш ответ на вопрос