一尘不染

检测序列参数的正确方法?

python

我想编写一个接受参数的函数,该参数可以是序列或单个值。值的类型为str,int等,但我
希望将其限制为硬编码列表。换句话说,我想知道参数X是一个序列还是必须转换为序列的对象,以避免以后出现特殊情况。我可以做

type(X) in (list, tuple)

但是可能还有我不知道的其他序列类型,也没有通用的基类。

-N。

编辑 :请参阅下面的“答案”,了解其中大多数答案对我没有帮助的原因。也许您有更好的建议。


阅读 121

收藏
2020-12-20

共1个答案

一尘不染

上述所有方式的问题在于,str被视为一个序列(可迭代,具有 getitem 等),但通常被视为单个项目。

例如,一个函数可以接受一个参数,该参数可以是文件名或文件名列表。函数从后一个检测第一个的最Pythonic方法是什么?

根据修订后的问题,听起来您想要的是更多类似的东西:

def to_sequence(arg):
    ''' 
    determine whether an arg should be treated as a "unit" or a "sequence"
    if it's a unit, return a 1-tuple with the arg
    '''
    def _multiple(x):  
        return hasattr(x,"__iter__")
    if _multiple(arg):  
        return arg
    else:
        return (arg,)

>>> to_sequence("a string")
('a string',)
>>> to_sequence( (1,2,3) )
(1, 2, 3)
>>> to_sequence( xrange(5) )
xrange(5)

不能保证可以处理 所有 类型,但是可以很好地处理您提到的情况,并且应该对大多数内置类型都适用。

使用它时,请确保接收此输出的任何人都可以处理可迭代对象。

2020-12-20