我有一个函数,其输入参数可以是元素或元素列表.如果这个参数是一个单一的元素,那么我把它放在列表中,所以我可以一致地迭代输入.

目前我有这个:

def my_func(input):
    if not isinstance(input,list): input = [input]
    for e in input:
        ...

我正在使用现有的API,因此我无法更改输入参数.使用isinstance()感觉到hacky,所以有一个正确的方法来做到这一点?

解决方法

我喜欢Andrei Vajna关于hasattr(var,’__ iter__’)的建议.注意这些典型的Python类型的结果:

>>> hasattr("abc","__iter__")
False
>>> hasattr((0,),"__iter__")
True
>>> hasattr({},"__iter__")
True
>>> hasattr(set(),"__iter__")
True

这具有将字符串视为不可迭代的附加优点 – 字符串是灰色区域,有时您希望将它们视为元素,而另一些作为字符序列.

请注意,在Python 3中,str类型确实具有__iter__属性,这不起作用:

>>> hasattr("abc","__iter__")
True

dawei

【声明】:唐山站长网内容转载自互联网,其相关言论仅代表作者个人观点绝非权威,不代表本站立场。如您发现内容存在版权问题,请提交相关链接至邮箱:bqsm@foxmail.com,我们将及时予以处理。