Построение с NaNs. Как установить значение NaN для определенного цвета и / или пропустить NaN из тепловой карты

Question

Я пытаюсь построить тепловую карту, используя патчи. Однако данные содержат NaN и устанавливают все цвета на минимальное значение, если включены значения NaN. Если я установлю NaNs на 0, график проиллюстрирован правильно. Но я не хочу устанавливать значение NaN, так как это вводит в заблуждение. В идеале я хотел бы пропустить значения NaN, если не установить хотя бы желаемый цвет

. Я обязательно использовал nanmin / nanmax для цветной полосы, чтобы исключить значения NaN из шкалы.

При построении графика от -2000: 2000 по 400 элементам я установил значения индекса 200: 250 в NaN

Приведенный выше код был использован для получения ниже: https://www.pastepic.xyz/images/2019/10/17/image9347fc65ec10705e.png

Однако включение NaNS приводит к этому https://www.pastepic.xyz/images/2019/10/17/imagef0c34416d84b669d.png

Answer 1

Эй, я не вижу ваш код, поэтому я предполагаю, что вы используете numpy, и тогда мы можем просто удалить nans из данных, например, если данные одномерные

x = x[~numpy.isnan(x)]

Мы также можемиспользуйте маску для удаления nans следующим образом, если данные имеют x и y.

import numpy as np
import matplotlib.pyplot as plt

x = np.arange(10)
y = np.array([1, 4, 3, None, None, 5, 8, 9, 4, 7])
mask = np.isfinite(y)

Тогда данные для x и y станут

x = x[mask] 
y = y[mask]

, дайте мне знать, как это работает для вас