我希望一个函数能够将字符串列表或多个字符串作为参数作为 *args

Question

我有一个函数，它应该能够将许多字符串参数作为 *args，或者将字符串列表作为参数。例如：

def getStuff(*stuff):
  for thing in stuff:
    print(thing)
getStuff("cat", "mouse", "dog")
getStuff(animals)

如果我调用它，我希望这个函数能够产生相同的结果。我目前正在使用以下非常简单的方法，但不能生成最干净的代码：

def getStuff(*stuff):
  if type(stuff[0]) != list:
    for thing in stuff:
        print(thing)
  else:
    for thing in stuff:
      for subthing in thing:
        print(subthing)

有没有一种简单的方法可以做到这一点？我正在寻找 python 最佳实践。

Answer 1

在 Python 中，许多人更喜欢遵循EAFP 原则而不是类型检查（又名LBYL）——因为异常处理相当便宜——请参阅What is the EAFP principle in Python? 特别是this answer。

以下是如何将其应用于您的示例代码：

def getStuff(*stuff):
    try:
        stuff[0].split()
    except AttributeError:  # List objects have no split() method.
        stuff = stuff[0]
    for thing in stuff:
        print(thing)

getStuff("cat", "mouse", "dog")
print()
animals = ['cow', 'horse', 'pig']
getStuff(animals)

输出：

cat
mouse
dog

cow
horse
pig

Answer 2

如果它是一个列表，这将获取 args 元组的第一个元素，否则我们可以循环遍历 args (stuff) 元组本身：

def getStuff(*stuff):
    
    stuff = stuff[0] if isinstance(stuff[0], list) else stuff
    
    for thing in stuff:
        print(thing)

更优雅的解决方案，使用itertools:

import itertools

def getStuff(*stuff):
        
    for thing in itertools.chain.from_iterable(stuff):
        print(thing)

解释：itertools.chain.from_iterable 只是将嵌套的可迭代对象展平，以防stuff 不仅仅是一个字符串元组。像这样stuff是元组还是元组中的列表，甚至是多个列表的元组都没有关系。

Answer 3

在 Python 中编写函数时，通常最佳做法是假设函数的参数是已知且不变的。

查看您的代码，如果您无论如何都将字符串数组传递给您的函数，似乎可以完成同样的事情。将数组传递给它后，只需打印其中的每个项目。如果只有一个项目，它将与仅打印该单个项目的功能相同。

我建议这样写：

def getStuff(stuff):
    for thing in stuff:
        print(thing)

我确实意识到这并不是您想要的，但是当谈到 Python 中的最佳实践时，就是这样。