Построение собственного поведения с помощью matplotlib

Я пытаюсь построить собственное поведение с помощью matplotlib, в основном у меня есть 2D-матрица, и я хотел бы построить ее с помощью чего-то очень похожего на тепловую карту, но ячейки разделены и узнаваемы. См., например:

http://www.cl.cam.ac.uk/~nv240/pics/eigenbehaviour.jpg