Поймать numpy ComplexWarning как исключение ⇐ Python

[Anonymous]

Рассмотрим следующий пример:

Код: Выделить всё

>>> import numpy as np
>>> a = np.array([1.0, 2.1j])
>>> b = np.array(a, dtype=np.float64)
/Users/goerz/anaconda3/bin/ipython:1: ComplexWarning: Casting complex values to real discards the imaginary part
#!/Users/goerz/anaconda3/bin/python

Как я могу перехватить ComplexWarning как исключение?

Я пробовал np.seterr, но это не дает никакого эффекта (поскольку это относится только к предупреждениям с плавающей запятой, таким как недополнение/переполнение).

Я также пробовал контекстный менеджер with alerts.catch_warnings(): из стандартной библиотеки, но он тоже не имеет никакого эффекта.

Подробнее здесь: https://stackoverflow.com/questions/548 ... -exception