Состав Python-функции - PullRequest
Новая
       11

Состав Python-функции

9 голосов
/ 17 февраля 2010

Я попытался реализовать композицию функций с хорошим синтаксисом, и вот что я получил: 

from functools import partial

class _compfunc(partial):
    def __lshift__(self, y):
        f = lambda *args, **kwargs: self.func(y(*args, **kwargs)) 
        return _compfunc(f)

    def __rshift__(self, y):
        f = lambda *args, **kwargs: y(self.func(*args, **kwargs)) 
        return _compfunc(f)

def composable(f):
    return _compfunc(f)

@composable    
def f1(x):
    return x * 2

@composable
def f2(x):
    return  x + 3

@composable
def f3(x):
    return (-1) * x

print f1(2) #4
print f2(2) #5
print (f1 << f2 << f1)(2) #14
print (f3 >> f2)(2) #1
print (f2 >> f3)(2) #-5

Отлично работает с целыми числами, но не работает в списках / кортежах: 

@composable
def f4(a):
    a.append(0)

print f4([1, 2]) #None

Где ошибка?

1 Ответ

9 голосов
/ 17 февраля 2010

append делает сложение на месте, как сказал Игнасио Васкес-Абрамс (хорошо, подразумевается) - так что, хотя вы могли бы это исправить, просто добавив return к своей функции, он имел бы побочный эффект для изменения аргумента он тоже был передан: 

@composable
def f4(a):
    a.append(0)
    return a

Было бы лучше использовать следующий, еще более краткий код, который также создает и возвращает новый объект: 

@composable
def f4(a):
  return a + [0]
...