无缝处理可迭代和不可迭代答案

【问题标题】：Handle iterable and non-iterable seamlessly无缝处理可迭代和不可迭代
【发布时间】：2011-02-18 22:58:49
【问题描述】：

能否告诉我如何优化以下代码？

def f(y, list_or_elem):
  if getattr(list_or_elem, '__iter__'):
    y = max(y, *list_or_elem)
  else:
    y = max(y, list_or_elem)

【问题讨论】：

【解决方案1】：

最好的优化是避免将“列表或单个元素”作为参数这样的愚蠢行为。但是，如果您坚持，最好使用 try/except 尽快删除异常并确保可以迭代：

try: iter(list_or_elem)
except TypeError: iterable = [list_or_elem]
else: iterable = list_or_elem
y = max(y, *iterable)

【讨论】：

【解决方案2】：

如果您愿意在您的代码中添加 flatten 功能（有一个很好的 here），它基本上可以获取...的列表列表并将其归结为单个列表，您可以这样做像

y = max(flatten([y, list_or_elem]))

【讨论】：