python - 在每个单元格中绘制/绘制具有自定义颜色的棋盘

Question

我正在寻找一个 Python 函数来绘制/绘制具有自定义数量的单元格和每个单元格中的自定义颜色的棋盘。该脚本可以基于matplotlib我实际上更喜欢的 .

如果我处理颜色而不是热图，那就太好了。例如，如果我可以像这样使用这个函数，那就太好了：

# Coding in Python:
#chess[row,col] = color;
chess[0,0] = yellow;
chess[0,3] = black;

类似于以下内容：

如果实现是参数化的，那就太好了。

标签： pythonmatplotlibplot

Answer 1

您需要定义颜色的 RGB 值，然后使用imshow

例如：

import numpy as np
yellow=(255,255,0)
red=(255,0,0)
blue=(0,0,255)
board=np.array([[yellow,red,blue],[blue,yellow,red]])
plt.imshow(board)

将产生：