Python強大的可變參數傳遞機制

今天模擬定義map函數.寫着寫着就發現Python可變長度參數的機制真是靈活而強大.

假設有一個元組t,包含n個成員:

t=(arg1,...,argn)

而一個函數f恰好能接受n個參數:

f(arg1,...,argn)

f(t)這種做法顯然是錯的,那么如何把t的各成員作為獨立的參數傳給f,以便達到f(arg1,...,argn)的效果?

我一開始想到的是很原始的解法,先把t的各個成員變為字符串的形式,再用英文逗號把它們串聯起來,形成一個"標准參數字符串":

str_t=(str(x) for x in t)
str_args=",".join(str_t)

str_args成了字符串"arg1,...,argn",這樣,

eval('%s(%s)'%(f.__name__,str_args))

看起來就相當於:

f(arg1,...,argn)

 

 

老版本:

def imap(func,arr,*arrs):
    allarrs=(arr,)+arrs
    args=[]
    for i in range(len(arr)):
        member=[]
        for ar in allarrs:
            member.append(str(ar[i]))
        args.append(member)
    return (eval('%s(%s)'%(func.__name__,','.join(member))) for member in args)

print list(imap(float,(1,2,3,4)))
print list(imap((lambda x,y,z:x+y+z),(1,1,1,1),(1,1,1,1),(1,1,1,1)))

一運行發現,對於命名函數float可以正常運作,但對於匿名函數lambda就不行了.顯然eval思路的局限就在這里.

我突然想起,直接f(*t) 就可以達到目的了!於是,新版本:

def imap(func,arr,*arrs):
    allarrs=(arr,)+arrs
    return (func(*(ar[i] for ar in allarrs)) for i in range(len(arr)))

print list(imap(float,(1,2,3,4)))
print list(imap((lambda x,y,z:x+y+z),(1,1,1,1),(1,1,1,1),(1,1,1,1)))

結果:

>>> 
[1.0, 2.0, 3.0, 4.0]
[3, 3, 3, 3]

而且function(*args_tuple)的強大機制在於,args_tuple不一定是元組,任意可迭代對象均可.列表,字典,生成器等.

>>> def function(*iterable):
    print iterable

    
>>> function(*(1,2,3))
(1, 2, 3)
>>> function(*[1,2,3])
(1, 2, 3)
>>> function(*{1:'',2:''})
(1, 2)
>>> function(*(i for i in range(4)))
(0, 1, 2, 3)
>>>