Matplotlib：三種顏色的自定義顏色圖

Question

我想通過一個顏色映射到matplotlib.imshow來繪製一個NxN數組，其中每個單元格只能取值爲-1,0和1. 我希望我的顏色映射恰好映射到一個顏色（十六進制代碼）到每個值。

我只是不知道如何定義不是兩種顏色之間的漸變的自定義顏色映射。

Answer 1

提供了一個非常類似問題的解決方案here。總之，colors.ListedColormap(list-of-colours)，是你的朋友。

Answer 2

找到了答案，我的問題，因爲費爾南多建議：

import matplotlib 
colorsList = [(tuple rgb color 1),(tuple rgb color 2),(tuple rgb color 3)] 
CustomCmap = matplotlib.colors.ListedColormap(colorsList)