如何在 Python 中确定变量是 int 还是 list for 循环

Question

我想编写一个函数，它接受一个 int 或一个 int 列表作为参数并处理该值。

类似的东西：

def func(input):
  if isinstance(input, int):
    print(input)
  else:
    for i in input:
      print(i)

这对我来说看起来不是很pythonic，但我不知道如何做得更好。有没有办法在 int 上“循环”或将 int “打包”到处理这两个选项的列表中

谢谢

Answer 1

你也可以遵循pythonic原则EAFP（请求原谅比请求许可更容易）

def func(your_input):
    try:
        for i in your_input:
            print(i)
    except TypeError:
        print(your_input)

您尝试迭代 input 中的条目并打印它们。如果输入类型不可迭代（如 int），则尝试迭代不可迭代对象会引发 TypeError 异常。在except 子句中，您可以打印输入。

编辑：正如@buran 所说，input 是 Python 的保留关键字，因此您应该避免使用它。

Answer 2

我不会太担心它是否是 Pythonic。但是这是另一种方法，它不会限制您使用列表或整数，而是检查您是否有可迭代的：

def func(input):
    if not hasattr(input, '__iter__'):
        input = [input]

    for i in input:
        print(i)

但要考虑的另一个可能更清洁的选择可能是：

def func(*inputs):
    for i in inputs:
        print(i)

这个函数的调用方式略有不同：func(5) 用于标量，func(*lst) 用于迭代。

Answer 3

为了完整起见，除了@jcf 答案，您还可以使用functoools.sigledispatch（或functools.singledispatchmethod，如果在课堂上） 它允许重载具有不同实现的函数，具体取决于第一个参数的类型

from functools import singledispatch

@singledispatch
def func(foo):
    print(foo)

@func.register(list)
def _(foo):
    for item in foo:
        print(item)

func(1)
func([2, 3, 4])

现在的问题是，如果你得到不同于列表的东西，比如元组或生成器等，你会怎么做。你可能想要使用来自 collections.abc 的抽象类，而不是 list

附带说明 - 不要使用 input 作为名称 - 这是一个内置函数。

Answer 4

def func(input):
    for i in ([input] if isinstance(input, int) else input):
        print(i)

根据需要将 int “打包”到列表中，但我认为它不会比您的原始代码更优雅。

Answer 5

if type(input) == int:
    print('Integer')
    # Do integer operation
elif isinstance(input, list):
    print('List')
    # Do list operation