Python函數參數：元組/列表

Question

我的函數需要一個列表或元組作爲參數。它並不真正關心它是，它所做的只是把它傳遞給可以接受一個列表或元組的另一個功能：

def func(arg): # arg is tuple or list 
    another_func(x) 
    # do other stuff here 

現在我需要稍微修改的功能，來處理額外的元素：

def func(arg): #arg is tuple or list 
    another_func(x + ['a']) 
    # etc 

不幸的是，這是行不通的：如果arg是元組，我必須說x + ('a',)。

很明顯，我可以通過強制arg列表工作。但它並不整齊。

有沒有更好的方法呢？當然，我不能強迫呼叫者總是通過一個元組，因爲它只是轉移到他們工作。

Answer 1

如果another_func只是想要一個迭代你可以通過itertools.chain(x,'a')它。

Answer 2

怎麼樣：

l = ['a'] 
l.extend(x) 

編輯： 重讀的問題，我覺得這是更你想要什麼（使用arg和x是有點混亂）：

tuple(arg) + ('a',) 

正如其他人所說，這可能不是最有效的方式，但它非常清楚。如果你的元組/列表很小，我可能會使用這個不太清晰的解決方案，因爲性能影響可以忽略不計。如果您的清單很大，請使用更高效的解決方案。

Answer 3

如何將其他函數改爲接受params列表呢？

def func(arg): # arg is tuple or list 
    another_func('a', *x) 

Answer 4

def f(*args): 
    print args 

def a(*args): 
    k = list(args) 
    k.append('a') 
    f(*k) 

a(1, 2, 3) 

輸出：

(1, 2, 3, 'a') 

Answer 5

我的建議：

def foo(t): 
    bar(list(t) + [other]) 

這是不是很有效，但是，你會更好繞過可變的東西，如果你要成爲，好吧，改變它們。

Answer 6

讓你的函數接受任何迭代。然後使用itertools.chain將您想要的任何序列添加到迭代器中。

from itertools import chain 

def func(iterable): 
    another_func(chain(iterable, ('a',))) 

Answer 7

可以使用迭代傳遞給第一個函數的類型來構建你傳遞給第二個內容：

from itertools import chain 

def func(iterable): 
    it = iter(iterable) 
    another_func(type(iterable)(chain(it, ('a',)))) 

def another_func(arg): 
    print arg 

func((1,2)) 
# (1, 2, 'a') 
func([1,2]) 
# [1, 2, 'a'] 

Answer 8

如果一個迭代是的話，你可以使用itertools.chain，但要知道，如果函數A（第一個被調用）在調用B後迭代遍歷該迭代，那麼您可能會遇到問題，因爲無法重新迭代迭代。在這種情況下，你應該選擇一個序列或使用iterable.tee使可迭代的副本：

import itertools as it 

def A(iterable): 
    iterable, backup = it.tee(iterable) 
    res = B(it.chain(iterable, 'a')) 
    #do something with res 
    for elem in backup: 
     #do something with elem 

def B(iterable): 
    for elem in iterable: 
     #do something with elem 

即使itertools.tee是不是真的那麼有效的，如果B消耗所有或最可迭代的，那麼只需將iterable轉換爲tuple或list即可。

Answer 9

我會說這樣做

def foo(t): 
    bar(list(t) + [other]) 

聖地亞哥Lezica的回答是最好的，因爲它是最簡單的。 （不需要導入itertools的東西，並使用更少的可讀鏈式調用）。但只有在你希望t很小的時候才能使用它。如果t可能很大，您應該使用其他解決方案之一。