matplotlibのカラーマップで表示された色によって透過処理したい

Question

2次元行列（atte_maps[0]）をmatplotlibによってカラーマップとして保存しています。

plt.figure('atte_map_1')
plt.imshow(atte_maps[0][0, :, :, 0], cmap='jet')
plt.axis('off')
plt.savefig(output_path+'/atte_map_1.png', bbox_inches='tight',pad_inches = 0)

出力される画像は以下のようなものです。

例えば赤〜オレンジ〜黄色のような色は白、それ以外の範囲は透明であるRGBA画像にうまく変換する方法はあるでしょうか？

色の範囲指定のやり方がよくわからず困っています。
matplotlib以外のライブラリを使うことも検討しています。ご教示いただけますと幸いです。

各バージョンは以下を使用
matplotlib == 3.2.2
python == 3.6

Comments

[hoshi-takanori]

色は R, G, B の 3 次元の値なので、範囲指定は簡単ではないですね。
元のデータが各点についてスカラー値なら、それを直接白か透明にすれば良いのでは。

Answer 1

自己解決しました。
cmapをbinaryにしたところ、各ピクセルの色判定が簡単にできました。