Pythonic 方法来制作一个可以接受可迭代或任意数量参数的函数？

Question

这真的是最 Pythonic 的方式来制作一个可以接受一个可迭代或任意数量的参数并对每个参数执行某些操作的函数吗？

from collections.abc import Iterable


def for_each(*args):
    """Take any number of arguments or an iterable and do something with them.
    Shouldn't throw an error or print anything when no arguments are given.
    """
    if (len(args) == 1
            and isinstance(args[0], Iterable)
            and not isinstance(args[0], str)):
        # Is this really the most Pythonic way to do this?
        args = args[0]

    for arg in args:
        # heavily simplified
        print("done", arg)


for_each(1, 2)
for_each(['foo', 'bar'])
for_each((3, 4, 5))
for_each()  # does essentially nothing, like it should
for_each("string")  # don't loop over the characters
for_each(6)

输出：

done 1
done 2
done foo
done bar
done 3
done 4
done 5
done string
done 6

我得到了这个有效的答案from here，但由于我实际上是在寻找一种更清洁的方法来做到这一点，所以我提出了这个新问题。

这可行，但在我看来，检查非常难看，我希望有更简单、更清洁的方法来实现这一点。 输出应该保持不变。

Answer 1

Pythonic 的方式是不要。选择一个，让调用者提供正确的参数：

# Iterable only: caller wraps the arguments
def for_each(iterable=None):
    if iterable is None:
        iterable = ()
        # or just return now, if that's an option

    for arg in iterable:
        ...

foreach((1,2,3))

# separate args: caller unpacks the arguments
def for_each(*args):
    for arg in args:
        ...

x = [1,2,3]
for_each(*x)

没有什么好方法可以随心所欲，因为你天生就试图猜测调用者的意图。您必须计算参数的数量，然后您必须担心像 str 这样的类型看起来像可迭代但不是，或者它们应该是。

如果必须按照您想要的方式完成，您的原始示例几乎可以做到，但它仍然存在缺陷。假设用户希望将字符串视为可迭代;您的方法仍然需要他们编写 for_each(["foo"])，在这种情况下，为什么无缘无故地使您的函数实现复杂化。