函數樣可變

Question

[由this question啓發]

假設我有兩個列表：

list1 = ['tom', 'mary', 'frank', 'joe', 'john', 'barry'] 
list2 = [1, 2, 3, 4] 

我想每名list1與list2一個數量相匹配。由於list2中有四個數字，因此我想將前四個名稱中的每一個與list2中的相應數字進行配對，並將隨機數字分配給list1中的其餘名稱。

我知道我可以使用for循環與enumerate和random.choice解決此問題。事實上，我提供了such a solution來解決原始問題。
我想不過要知道，如果有可能做這樣的事情：

for name, number in itertools.izip_longest(list1, list2, fillvalue=MAGIC): 
    print name, number 

本來，我想用這樣的事情的：

MAGIC = random.choice(list1) 

但是，執行random.choice(list1)第一，然後將該答案用作壓縮操作的fillvalue。這是沒有吸引力的，因爲它不會爲每一對拉值選擇一個新的隨機值。因此很明顯，itertools.izip_longest要求它的fillvalue，它本身具有價值，它不稱之爲。事實上，如果我要提供一個函數，它將產生一個由名稱和可調用函數組成的對，這也是不足的。出於這個原因，lambda函數是不可行的解決方案。

我該如何去創建一個變量來調用某個函數？ itertools.izip_longest如何使用fillvalue變量？該變量的__repr__被稱爲？如果是這樣，我可以用__repr__做一個類，它在裏面調用一個函數嗎？

Answer 1

在這裏似乎最簡單的方法是創建一個發生器產生的所有永恆的填充值，然後在列表2屈服值連鎖這樣的：

def inf_gen(f): 
    while True: 
     yield f() 

注意inf_gen(f)實際上相當於iter(f, None)，因爲f永遠不會返回。帶有兩個參數的iter調用第一個參數，直到它返回第二個參數。通過使用一臺發電機理解與itertools.count

>>> from itertools import izip, chain 
>>> for name, number in izip(list1,chain(list2, iter(lambda: random.choice(list2), None))): 
    print name, number 

tom 1 
mary 2 
frank 3 
joe 4 
john 1 
barry 4 

你也可以這樣做只是itertools的東西，而定義一個單獨的功能：如果你想讓它

>>> from itertools import izip, chain, count 
>>> for name, number in izip(list1, chain(list2, (random.choice(list2) for _ in count()))): 
    print name, number 

Answer 2

然後使用它是這樣爲了「縮放」到多個迭代器，您可以通過對http://docs.python.org/2.7/library/itertools.html?highlight=izip_longest#itertools.izip_longest中給出的代碼稍作修改來完成：

from itertools import repeat, chain, count 

class ZipExhausted(Exception): 
    pass 

def izip_longest2(*args, **kwds): 
    fillvalue = kwds.get('fillvalue') 
    fillgen = kwds.get('fillgen', repeat(fillvalue)) # added 
    counter = [len(args) - 1] 
    def sentinel(): 
     if not counter[0]: 
      raise ZipExhausted 
     counter[0] -= 1 
     yield next(fillgen) # modified 
    iterators = [chain(it, sentinel(), fillgen) for it in args] # modified 
    try: 
     while iterators: 
      yield tuple(map(next, iterators)) 
    except ZipExhausted: 
     pass 


a = ['bob', 'mary', 'paul'] 
b = ['eggs', 'spam'] 
c = ['a', 'b', 'c', 'd'] 

for x in izip_longest2(a, b, c, fillgen=count()): 
    print '\t'.join(map(str, x)) 

# output: 
# bob eggs a 
# mary spam b 
# paul 0 c 
# 1 2 d