首页
学习
活动
专区
工具
TVP
发布
社区首页 >问答首页 >我如何捕捉一个numpy警告,就像它是一个异常一样(不仅仅是为了测试)?

我如何捕捉一个numpy警告,就像它是一个异常一样(不仅仅是为了测试)?
EN

Stack Overflow用户
提问于 2013-04-11 02:36:22
回答 2查看 148.3K关注 0票数 213

我必须在Python中创建一个拉格朗日多项式,用于我正在做的项目。我正在做一个重心风格的循环,以避免使用显式的for循环,而不是牛顿的除差风格的循环。我遇到的问题是,我需要捕获一个被零除的结果,但是Python (或者numpy)只是把它变成了一个警告,而不是一个正常的异常。

因此,我需要知道如何捕捉这个警告,就像它是一个异常一样。我在这个网站上发现的与此相关的问题没有以我需要的方式得到回答。下面是我的代码:

代码语言:javascript
复制
import numpy as np
import matplotlib.pyplot as plt
import warnings

class Lagrange:
    def __init__(self, xPts, yPts):
        self.xPts = np.array(xPts)
        self.yPts = np.array(yPts)
        self.degree = len(xPts)-1 
        self.weights = np.array([np.product([x_j - x_i for x_j in xPts if x_j != x_i]) for x_i in xPts])

    def __call__(self, x):
        warnings.filterwarnings("error")
        try:
            bigNumerator = np.product(x - self.xPts)
            numerators = np.array([bigNumerator/(x - x_j) for x_j in self.xPts])
            return sum(numerators/self.weights*self.yPts) 
        except Exception, e: # Catch division by 0. Only possible in 'numerators' array
            return yPts[np.where(xPts == x)[0][0]]

L = Lagrange([-1,0,1],[1,0,1]) # Creates quadratic poly L(x) = x^2

L(1) # This should catch an error, then return 1. 

当执行这段代码时,我得到的输出是:

代码语言:javascript
复制
Warning: divide by zero encountered in int_scalars

这就是我想要捕捉到的警告。它应该发生在列表理解内部。

EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/15933741

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档