imshowの画像の色を値で指定したい

Question

実現したいこと

imshowの画像の色を値で指定したいです。

前提

発生している問題・エラーメッセージ

該当のソースコード

import numpy as np
from matplotlib import pyplot as plt

d = np.array([15,15,30,200,30,0,40,90,0])
d1 = d.reshape(3, 3)
fig, ax = plt.subplots()
im = ax.imshow(d1,cmap="coolwarm")
plt.colorbar(im)

試したこと

値が0なら　白　value=0 ,color='white'
値が20未満なら　青　value < 20 ,color='cyan'
値が20以上50未満なら　緑　value =>20 & <50 ,color='green'
値が50以上なら　赤　value >50 ,color='red'

を、cmap=""の部分に追加したらよいかと思い、
いろいろ調べてみましたが、よくわかりませんでした。

ネット（https://tech-market.org/matplotlib-cmap/）でしらべると、
「画像のcmapを変更したい場合はplt.imshow()のcmapにカラーマップ名を与えることで変更が可能」
とありました。すでにあるカラーマップだとよいのですが、良いものが見当たりません。
「自分でカラーマップを作成することもできる」（https://disassemble-channel.com/matplotlib-colormap-cmap/）という情報をみつけましたが、数値を色に変換する具体的なコードの書き方について読み取ることができませんでした。

どうぞよろしくお願いいたします。

補足情報（FW/ツールのバージョンなど）

Pythonは　3.10.12　です。

Comments

[meg_]

・コードは「コードの挿入」で記入してください。

> いろいろ調べてみましたが、よくわかりませんでした。
重複回答を避けるために具体的に教えていただけますか？調べたサイトや具体的に「よくわかりませんでした」の内容等。

[ria_1965]

コードは「コードの挿入」に記入することと、重複回答を避けるために、調べたサイトやわからないことの具体的な内容について、質問内容を編集しました。

Answer 1

matplotlib.colors.BoundaryNorm — Matplotlib 3.7.2 documentation を利用する場合。

import numpy as np
from matplotlib import pyplot as plt
from matplotlib.colors import ListedColormap, BoundaryNorm

d = np.array([15,15,30,200,30,0,40,90,0])
d1 = d.reshape(3, 3)
fig, ax = plt.subplots()

#
colors = ['white', 'cyan', 'green', 'red']
cmap = ListedColormap(colors)
boundary = np.array([0, 1, 20, 50, np.max(d)])
norm = BoundaryNorm(boundary, ncolors=len(colors))
im = ax.imshow(d1,cmap=cmap,norm=norm)

plt.colorbar(im, spacing='proportional')
plt.show()

Comments

[ria_1965]

ありがとうございました！