方法中的单个字符串或字符串列表[重复]

Question

我已经遇到过好几次了。我正在处理很多可以接受字符串列表的方法。有几次我不小心传递了一个字符串，它被分解成一个列表，每个字符都被使用，这不是我们想要的行为。

def test(a,b):
    x = []
    x.extend(a)
    x.extend(b)
    return x

x = [1,2,3,4]

我不想发生的事情：

test(x,'test')
[1, 2, 3, 4, 't', 'e', 's', 't']

我不得不求助于一种奇怪的语法：

test(x,['list']) 

我希望这些隐式工作：

test(x,'list')
[1, 2, 3, 4, 'test']

test(x,['one', 'two', 'three'])
[1, 2, 3, 4, 'one', 'two', 'three']

我真的觉得有一种“pythonic”的方法可以做到这一点或涉及鸭子打字的东西，但我没有看到。我知道我可以使用 isinstance() 来检查它是否是一个字符串，但我觉得有更好的方法。

编辑：我使用的是 python 2.4.3

Answer 1

使用这个

def test( x, *args ):

现在你可以做

test( x, 'one' )

和

test( x, 'one', 'two' )

和

test( x, *['one', 'two',] )

Answer 2

我很抱歉这么说，但这确实是使用isinstance() 有意义的少数情况之一。然而，建议对 list 进行测试的其他答案之一是倒退：str（以及 Python 2.x 的 unicode）是异常类型，因此您要测试这些类型：

def test(a,b):
    x = []
    if isinstance(b, str): b = [b]
    x.extend(a)
    x.extend(b)
    return x

或用于 Python 2.x 测试 isinstance(b, basestring)。

Answer 3

来自The Zen of Python：

Explicit is better than implicit.

在我看来，你的第一个例子是明确的。它采用两个值并以易于理解的方式处理它们。尽管对于新的 Python 程序员来说它“感觉”有点奇怪，但它的行为就像期望一样。 list.extend 接受列表，因此它将字符串视为字符列表。

您的提议将list.extend 的语义更改为等同于list.append，但仅在处理字符串时。对于那些希望你的函数将你的字符串视为字符列表并且不得不像test(x,list('test')) 这样调用它的人来说，这将是一个真正的惊喜。

您可以按照您的要求做，但请不要这样做。 :-)

Answer 4

嗯，这很简单：

>>> def a(*args):
...     x=[]
...     for i in args:
...             if '__iter__' in dir(i):
...                     x+=list(i)
...             else:
...                     x.append(i)
...     return x
... 
>>> a(1,2,3,4)
[1, 2, 3, 4]
>>> a(1,2,[3,4])
[1, 2, 3, 4]
>>> a(1,2,[3,4],'123')
[1, 2, 3, 4, '123']
>>> a(1,2,[3,4],'123', 1231, (1, 2, 3, 4))
[1, 2, 3, 4, '123', 1231, 1, 2, 3, 4]

Answer 5

对 pod2metra 的回答稍作调整，即可使该函数能够处理所有级别的湮灭：

>>> def b(*args):
...     x=[]
...     for i in args:
...             if '__iter__' in dir(i):
...                     x += b(*i)
...             else:
...                     x.append(i)
...     return x
... 
>>> b(1, (2, (3, 4)))
[1, 2, 3, 4]

你有函数'a'：

>>> a(1, (2, (3, 4)))
[1, 2, (3, 4)]

我知道这不是最初问题的意图，但也许这种概括可能有用。

Answer 6

简单测试参数的类型：

def test(a,b):
    x = a if isinstance(a, list) else [a]
    x += b if isinstance(b, list) else [b]
    return x