python - Numpy/Polyfit - 禁止打印英特尔 MKL 错误消息

Question

我在一个程序中多次计算 polyfit，我的一些输入是 np.nan 并且会得到算法问题。我知道这一点，在这个应用程序中我不在乎。

当事情搞砸时，这会打印到控制台:

Intel MKL 错误:进入 DELSD 时参数 4 不正确。

我只是想抑制这个错误。我已经尝试过:

import warnings
warnings.simplefilter('ignore', np.RankWarning)
warnings.simplefilter('ignore', np.ComplexWarning)
warnings.filterwarnings('ignore', "Intel MKL ERROR")

它会抑制一些警告，但不会抑制 Intel MKL 警告。我只是想阻止它在控制台中打印(因为它会破坏我正在打印的其他状态消息)。

以下应该触发问题:

import numpy as np
def line_fit(R, X):
    num_rows = np.shape(R)[0]
    p = np.zeros(num_rows)
    for i in range(num_rows):
        temp = np.polyfit(R[i, :], X[i, :], 1)
        p[i] = temp[1]
    return p
temp = np.array((((198.652-76.1781j),(132.614-43.8134j),(115.042-41.2485j),(91.7754-39.1649j),(78.8538-37.389j),(67.8769-34.6342j)),
((np.nan),(1671.79-796.522j),(1206.44-824.202j),(654.572-682.673j),(438.175-559.025j),(303.624-452.122j)),
((np.nan-1j*np.nan),(1671.32-794.931j),(1198.71-803.533j),(649.574-624.276j),(443.286-530.36j),(308.609-438.738j))))
R = np.real(temp)
X = np.imag(temp)
coeff = line_fit(R, X)

Python 2.7.6(默认，2013 年 11 月 10 日，19:24:24)[MSC v.1500 64 位 (AMD64)]，NumPy 1.8.0

Answer 1

错误

Intel MKL ERROR: Parameter 4 was incorrect on entry to DELSD

当您的输入中有 Nan 或 Inf 值时发生。请检查并估算它。