在Python中用自定义颜色绘制矩阵

Question

我试图用python中的pytlot绘制一个矩阵。在

import matplotlib.pyplot as plt
m = [
    [1, 0, 2, 0, 0],
    [1, 1, 1, 2, 0],
    [0, 4, 1, 0, 0],
    [0, 4, 4, 1, 2],
    [1, 3, 0, 0, 1],
]

plt.imshow(m)

plt.show()

这看起来像我需要的，但问题是我需要能够为一个位置选择颜色。在

例如：

位置1x2必须为红色。在

要用Python和Pyplot归档吗？在

Answer 1

我想matplotlib.colors.ListedColormap就是您在这里需要的：

import matplotlib
import matplotlib.pyplot as plt

colors = 'lime red blue magenta yellow'.split()
cmap = matplotlib.colors.ListedColormap(colors, name='colors', N=None)

m = [
    [1, 0, 2, 0, 0],
    [1, 1, 1, 2, 0],
    [0, 4, 1, 0, 0],
    [0, 4, 4, 1, 2],
    [1, 3, 0, 0, 1],
]

plt.imshow(m, cmap=cmap)
plt.show()

如果需要更多的控制，也可以在列表中使用（R，G，B）元组（而不是预定义的颜色字符串）