Как игнорировать предупреждения Python для комплексных чисел

Question

У меня есть список, который должен включать как реальные значения, так и сложные значения. Однако, когда я не инициализировать массив, используя numpy.zeros (DTYPE = комплексные), я получаю следующее сообщение об ошибке:

n[2] = 3.7095 + 0.007j 

TypeError: can't convert complex to float

Однако, когда я установить DTYPE к сложному, я получаю следующее предупреждение, когда я использую реальное число (которое не нарушает мой код, но это вызывает массу ненужных красных предупреждающих сообщений, которые загромождать мою консоль:

ComplexWarning: Casting complex values to real discards the imaginary part

Как я могу отключить предупреждающее сообщение, или установить его так что значения могут быть мнимыми или сложными?

Answer 1

Вы можете использовать это, чтобы игнорировать предупреждающие сообщения:

import warnings 
warnings.filterwarnings('ignore') 

Документация: https://docs.python.org/2/library/warnings.html

Я включил ссылку на документацию, есть и более «мягкие» методы, чтобы иметь дело с предупреждением, например, печать только первого появления предупреждения («один раз»). В вашем случае также можно фильтровать только определенные классы предупреждений (ComplexWarning).