python heatmapの色操作:所定値の場合色変更できないでしょうか

Question

matplotlibやseabornを使ってheatmapを使ってデータの可視化を行っているのですが、思うように色操作ができず質問投稿しました。
具体的には、↓のグラフで数値0の場合(ピンク色の部分)を特別に指定色(白色)にしたいのですが、方法はないでしょうか。

maze = np.zeros([10,10])-10
maze[1:9,1:9]= 0

for idx in range(1,9):
    maze[idx,idx]=idx

figure      = plt.figure()
plt.imshow(maze, cmap="magma")
plt.colorbar()
plt.show()

Answer 1

matplotlib.colors.Colormap.set_bad を使う方法があります。

set_bad(color='k', alpha=None)

Set the color for masked values.

import numpy as np
import matplotlib
import matplotlib.pyplot as plt

maze = np.zeros([10,10])-10
maze[1:9,1:9]= 0

for idx in range(1,9):
    maze[idx,idx]=idx

figure      = plt.figure()

masked_maze = np.ma.masked_where(maze == 0, maze)
cmap = matplotlib.cm.magma
cmap.set_bad(color='white')

plt.imshow(masked_maze, cmap=cmap)
plt.colorbar()
plt.show()

Answer 2

【python】matplotlibでグラフに色を指定する方法【カラーマップ】やmatplotlib – カラーマップを自作する方法についてを見て、必要なカラーマップを自作して指定すれば可能です。