Tham số hàm Python: tuple/list

Question

Chức năng của tôi mong đợi một danh sách hoặc một bộ dữ liệu làm tham số. Nó không thực sự quan tâm mà nó là tất cả nó làm là vượt qua nó đến một chức năng chấp nhận hoặc là một danh sách hoặc tuple:

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

Bây giờ tôi cần phải thay đổi chức năng một chút, để xử lý một yếu tố bổ sung:

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

Thật không may điều này sẽ không hoạt động: nếu arg là tuple, tôi phải nói x + ('a',).

Rõ ràng, tôi có thể làm cho nó hoạt động bằng cách ép arg vào danh sách. Nhưng nó không gọn gàng.

Có cách nào tốt hơn để làm điều đó không? Tôi không thể ép buộc người gọi luôn luôn vượt qua một tuple, tất nhiên, vì nó chỉ đơn giản là chuyển sang làm việc với họ.

Answer 1

Nếu another_func chỉ muốn lặp lại, bạn có thể vượt qua itertools.chain(x,'a').

Answer 2

thế nào về:

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

Edit: Đọc lại câu hỏi, tôi nghĩ rằng đây là nhiều hơn những gì bạn muốn (việc sử dụng arg và x là một chút bối rối):

tuple(arg) + ('a',) 

Như những người khác đã nói, đây có lẽ không phải là cách hiệu quả nhất, nhưng nó rất rõ ràng. Nếu danh sách/danh sách của bạn nhỏ, tôi có thể sử dụng các giải pháp này ít hơn rõ ràng vì hiệu suất đạt được sẽ không đáng kể. Nếu danh sách của bạn lớn, hãy sử dụng các giải pháp hiệu quả hơn.

Answer 3

Điều gì sẽ thay đổi chức năng khác để chấp nhận danh sách các thông số thay thế?

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) 

Output:

(1, 2, 3, 'a') 

Answer 5

Đề nghị của tôi:

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

Đây không phải là rất hiệu quả tuy nhiên, bạn muốn được tốt hơn thông qua những thứ xung quanh có thể thay đổi nếu bạn đang đi để được , tốt, thay đổi chúng.

Answer 6

Chức năng của bạn có chấp nhận bất kỳ lần lặp nào không. Sau đó, sử dụng itertools.chain để thêm bất kỳ chuỗi nào bạn muốn lặp lại.

from itertools import chain 

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

Answer 7

Bạn có thể sử dụng các loại của iterable truyền cho hàm đầu tiên để xây dựng những gì bạn vượt qua thứ hai:

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

Nếu một iterable là đủ, bạn có thể sử dụng itertools.chain, nhưng lưu ý rằng nếu hàm A (cái đầu tiên được gọi), cũng lặp lại qua vòng lặp có thể lặp lại sau khi gọi B, thì bạn có thể gặp vấn đề vì các vòng lặp không thể được tua lại.Trong trường hợp này, bạn nên lựa chọn cho một chuỗi hoặc sử dụng iterable.tee để tạo một bản sao của iterable:

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 

Mặc dù itertools.tee là không thực sự là hiệu quả nếu B tiêu thụ tất cả hoặc nhất của iterable, tại thời điểm đó, đơn giản chỉ cần chuyển đổi iterable thành tuple hoặc list.

Answer 9

tôi muốn nói câu trả lời làm

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

Santiago Lezica là tốt nhất bởi vì nó là đơn giản nhất. (không cần phải nhập công cụ itertools và sử dụng các cuộc gọi chuỗi ít có thể đọc được nhiều hơn). Nhưng chỉ sử dụng nó nếu bạn mong đợi t được nhỏ. Nếu t có thể lớn, bạn nên sử dụng một trong các giải pháp khác.