Как я могу записать ценности NAN как специального цвета с imshow в Matplotlib?
-
24-09-2019 - |
Вопрос
Я пытаюсь использовать imshow в Matplotlib для сюжет данных как тепломапа, но некоторые значения являются назами. Я хотел бы, чтобы нан были оказаны как специальный цвет, не найден в сормере.
пример:
import numpy as np
import matplotlib.pyplot as plt
f = plt.figure()
ax = f.add_subplot(111)
a = np.arange(25).reshape((5,5)).astype(float)
a[3,:] = np.nan
ax.imshow(a, interpolation='nearest')
f.canvas.draw()
Результирующее изображение неожиданно все синие (самый низкий цвет в струйной колонне). Однако, если я сделаю так, как это так:
ax.imshow(a, interpolation='nearest', vmin=0, vmax=24)
- Затем я получаю что-то лучшее, но ценности NAN нарисованы одинаковым цветом, что и VMIN ... Есть ли изящный способ, которым я могу установить нан, чтобы быть нарисованным со специальным цветом (например: серый или прозрачный)?
Решение
HRM, кажется, я могу использовать маскируемый массив для этого:
masked_array = np.ma.array (a, mask=np.isnan(a))
cmap = matplotlib.cm.jet
cmap.set_bad('white',1.)
ax.imshow(masked_array, interpolation='nearest', cmap=cmap)
Это должно хватить, хотя я все еще открыт для предложений. :
Другие советы
С новыми версиями MATPLOTLIB не нужно больше использовать маскируемый массив.
Например, давайте генерируем массив, где каждое 7-е значение является NAN:
arr = np.arange(100, dtype=float).reshape(10, 10)
arr[~(arr % 7).astype(bool)] = np.nan
Мы можем изменить текущий сормер и построить массив со следующими строками:
current_cmap = matplotlib.cm.get_cmap()
current_cmap.set_bad(color='red')
plt.imshow(arr)