python - 迭代python中的列表或单个元素

Question

我想迭代一个未知函数的输出。不幸的是，我不知道该函数是返回单个项目还是一个元组。这一定是一个标准问题，并且必须有一个标准的方法来处理这个问题——我现在所拥有的非常丑陋。

x = UnknownFunction()
if islist(x):
    iterator = x
else:
    iterator = [x]

def islist(s):
    try:
        len(s)
        return True
    except TypeError:
        return False

for ii in iterator:
    #do stuff

Answer 1

这个问题最通用的解决方案是使用带有抽象基类collections.Iterable的isinstance。

import collections

def get_iterable(x):
    if isinstance(x, collections.Iterable):
        return x
    else:
        return (x,)

您可能还想测试 basestring 以及 Kindall建议。

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

现在有些人可能会想，就像我曾经做过的那样，“难道不是 isinstance considered harmful 吗？它不会将你锁定为使用一种类型吗？不会使用 hasattr(x, '__iter__') 会更好吗？"

答案是:对于抽象基类来说不是这样。实际上，您可以使用 __iter__ 方法定义自己的类，它会被识别为 collections.Iterable 的实例，即使您没有子类 collections.Iterable。这是因为 collections.Iterable 定义了 __subclasshook__它通过它实现的任何定义来确定传递给它的类型是否是 Iterable。

>>> class MyIter(object):
...     def __iter__(self):
...         return iter(range(10))
... 
>>> i = MyIter()
>>> isinstance(i, collections.Iterable)
True
>>> collections.Iterable.__subclasshook__(type(i))
True