如何处理列表推导式中的异常？

Question

我在 Python 中有一些列表理解，其中每次迭代都可能抛出异常。

例如，如果我有：

eggs = (1,3,0,3,2)

[1/egg for egg in eggs]

我会在第三个元素中收到 ZeroDivisionError 异常。

如何处理此异常并继续执行列表理解？

我能想到的唯一方法是使用辅助函数：

def spam(egg):
    try:
        return 1/egg
    except ZeroDivisionError:
        # handle division by zero error
        # leave empty for now
        pass

但这对我来说看起来有点麻烦。

在 Python 中是否有更好的方法来做到这一点？

注意：这是我设计的一个简单示例（参见上面的“例如”），因为我的真实示例需要一些上下文。我对避免除以零错误不感兴趣，但对处理列表理解中的异常感兴趣。

Answer 1

我意识到这个问题已经很老了，但是您也可以创建一个通用函数来使这种事情变得更容易：

def catch(func, handle=lambda e : e, *args, **kwargs):
    try:
        return func(*args, **kwargs)
    except Exception as e:
        return handle(e)

然后，在您的理解中：

eggs = (1,3,0,3,2)
[catch(lambda : 1/egg) for egg in eggs]
[1, 0, ('integer division or modulo by zero'), 0, 0]

您当然可以使默认句柄函数成为您想要的任何内容（假设您宁愿返回'默认情况下无）。

注意：在 python 3 中，我将仅使用“handle”参数关键字，并将其放在参数列表的末尾。这将使实际传递参数等通过 catch 变得更加自然。

更新（9年后...）：对于Python 3，我只是指切换*args和handle，这样你就可以指定函数的参数而不指定句柄。一个小便利：

def catch(func, *args, handle=lambda e : e, **kwargs):
    try:
        return func(*args, **kwargs)
    except Exception as e:
        return handle(e)

这在推导式中使用定义的函数时很有帮助：

from math import log    
eggs = [1,3,0,3,2]
[catch(log, egg) for egg in eggs]
[0.0, 1.0986122886681098, ValueError('math domain error'), 1.0986122886681098, 0.6931471805599453]

在 Python 2 版本下，我们必须在 egg 之前传入 handle 。

Answer 2

Python 中没有内置表达式可以让您忽略异常（或在异常情况下返回备用值 &c），因此从字面上讲，“在列表推导式中处理异常”是不可能的，因为列表推导式是一个包含其他表达式的表达式，仅此而已（即，没有语句，并且只有语句可以捕获/忽略/处理异常）。

函数调用是表达式，函数体可以包含您想要的所有语句，因此将容易出现异常的子表达式的计算委托给函数，正如您所注意到的，是一种可行的解决方法（其他可行的方法是检查可能引发异常的值，正如其他答案中所建议的那样）。

对“如何处理列表推导式中的异常”问题的正确回答都表达了这个事实的一部分：1）从字面上看，即在推导式本身的词汇上，你不能； 2）实际上，您可以将工作委托给一个函数，或者在可行的情况下检查容易出错的值。因此，您反复声称这不是答案是没有根据的。

Answer 3

您可以使用

[1/egg for egg in eggs if egg != 0]

它来简单地跳过零元素。

Answer 4

不，没有更好的办法了。在很多情况下，你可以像彼得一样使用回避

你的另一个选择是不使用推导式

eggs = (1,3,0,3,2)

result=[]
for egg in eggs:
    try:
        result.append(egg/0)
    except ZeroDivisionError:
        # handle division by zero error
        # leave empty for now
        pass

由你来决定这是否更麻烦

Answer 5

我认为，正如提出最初问题的人和 Bryan Head 所建议的那样，辅助函数很好，而且一点也不麻烦。一行神奇代码完成所有工作并不总是可能的，因此如果想避免 for 循环，辅助函数是一个完美的解决方案。不过我会将其修改为这个：

# A modified version of the helper function by the Question starter 
def spam(egg):
    try:
        return 1/egg, None
    except ZeroDivisionError as err:
        # handle division by zero error        
        return None, err

输出将是这个 [(1/1, None), (1/3, None), (None, ZeroDivisionError), (1/3, None), (1 /2，无）]。有了这个答案，您就可以完全控制以任何您想要的方式继续。

替代：

def spam2(egg):
    try:
        return 1/egg 
    except ZeroDivisionError:
        # handle division by zero error        
        return ZeroDivisionError

是的，返回错误，而不是引发错误。

Answer 6

我没有看到任何答案提到这一点。
但此示例是防止已知失败案例引发异常的一种方法。

eggs = (1,3,0,3,2)
[1/egg if egg > 0 else None for egg in eggs]


Output: [1, 0, None, 0, 0]

Answer 7

您可以使用生成器：

def invert(xs):
    for x in xs:
        try:
            yield x
        except:
            yield None

list(invert(eggs))