python - 在Python中返回满足特定条件的项目的最佳方法是什么

Question

我当前使用的代码是:

def f(*args):
    lst=[str(i) for i in args]
    if len(lst)==1:lst = lst[0]
    return lst

我想要的是:

a=f(1) #'1', not [1]

a,b = f(1,2) #'1', '2'

只有一个参数是列表，由 a 表示。

除了使用 if 语句之外，还有什么替代方法？

Answer 1

是的:

return lst[0] if len(lst) == 1 else lst