将列表或值传递给Python中的函数[重复]

Question

能够将列表或值传递给函数并将其返回为的最佳方法是什么

foo(x):foo(x)

foo([x, y, z]):[foo(x), foo(y), foo(z)]?

是否只是测试输入是否为列表？尝试访问列表元素的 try/except 块？还有什么？

Answer 1

使用集合来确定它是否可迭代。如果可迭代，请执行您的列表逻辑。如果不做你的变量逻辑

import collections

if isinstance(e, collections.Iterable) and not isinstance(e, basestring) :

正如 cmets 中指出的那样。添加逻辑以不包含字符串

Answer 2

这是一个简单的函数foo，它接受一个元素并将其乘以2。如果是列表，它将返回foo 操作的元素列表。

def foo(x):
    if isinstance(x, list): ## Accepts list ONLY
        return [foo(i) for i in x]
    return x * 2

输出：

>>> foo(3)
6
>>> foo([1,2,3])
[2, 4, 6]

编辑：

根据 Jon 的评论，如果您希望 tuple 成为可接受的输入，您可以执行以下操作：

def foo(x):
    if isinstance(x, (list, tuple)): ## Accepts list OR tuple
        return [foo(i) for i in x]
    return x * 2

不过，对于 list 或 tuple 输入，这仍将仅返回 list。如果你想要与输入相同类型的输出（即tuple 输出），那么你可以修改如下：

def foo(x):
    if isinstance(x, (list, tuple)): ## Accepts list OR tuple
        result = [foo(i) for i in x]
        return result if not isinstance(x, tuple) else tuple(result)
    return x * 2

新输出：

>>> foo(1)
2
>>> foo([1,2,3])
[2, 4, 6]
>>> foo((1,2,3))
(2, 4, 6)

还要注意，这将处理奇怪的嵌套结构：

>>> foo([[1,2,3],[4,5,6],(7,8,9)])
[[2, 4, 6], [8, 10, 12], (14, 16, 18)]