python - 迭代可能是迭代器或单个元素的东西

Question

假设我有以下功能:

def sum(summands)
    s = 0
    for a in summands:
        s = a + s

用户可能会使用列表 sum([1, 2, 3]) 来调用它，但如果您也可以直接使用数字 sum(5) 来调用它会很方便。 (这实际上与数字无关，只是一个简化的示例。)

我可以发明一个函数:

def make_iterable(x):
    # returns x if x is iterable, returns [x] if x is not iterable

但是是否有一种更短的内置方法可以使单个元素可迭代？

Answer 1

这个怎么样。

def sum(summands)
    s = 0

    try:
        iter(summands)
    except TypeError:
        return summands

    for a in summands:
        s = a + s
    return s

或者，如果您想使用您建议的 shell 函数，您可以将 try: except: 提取到 make_iterable

Python 2.x:

def make_iterable(x):
    try:
        iter(x)
    except TypeError:
        x=[x]
    return x

Python 3.x:

def make_iterable(x):
    try: yield from x
    except TypeError: yield x

然后用sum来调用

def sum(summands)
    s = 0

    summands = make_iterable(summands)

    for a in summands:
        s = a + s
    return s