Создание / Рисование шахматной доски с индивидуальным цветом в каждой ячейке

Question

Я ищу функцию Python для рисования / построения шахматной доски с индивидуальным количеством ячеек и индивидуальными цветами в каждой ячейке. Сценарий может быть основан на matplotlib, который я на самом деле предпочитаю.

Было бы замечательно, если бы я имел дело с цветами вместо тепловой карты. Например, было бы здорово, если бы я мог использовать эту функцию примерно так:

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

Что-то похожее на ниже:

Было бы здорово, если бы реализация была параметризована.

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)

даст: