在python中将具有不同值的矩阵可视化为颜色答案

【问题标题】：Visualize matrix with different values as colours in python在python中将具有不同值的矩阵可视化为颜色
【发布时间】：2020-03-08 12:26:26
【问题描述】：

我只想将具有三个不同值（例如 0、1 和 2）的矩阵打印为不同的颜色。想象一下我有以下网格

import numpy as np
grid=np.zeros((5, 5))
grid[1][1]=2
grid[3][8]=2
grid[3][4]=1
grid[6][7]=1
grid[7][7]=1

我怎样才能以一种像棋盘一样的方式将其可视化，并用一种颜色突出显示值“1”的位置，并用另一种颜色突出显示值“2”的位置。

我试过了

import matplotlib.pyplot as plt
ax=plt.subplots
ax.imshow(grid)

但我得到错误“'tuple' object has no attribute 'imshow'”

提前感谢您的帮助！

【问题讨论】：

【解决方案1】：

这对我有用：

grid=np.zeros((9, 9))
grid[1][1]=2
grid[3][8]=2
grid[3][4]=1
grid[6][7]=1
grid[7][7]=1

plt.imshow(grid)

请注意，您试图在示例中定义的数组之外设置一个元素。所以我把定义扩大到(9,9)

【讨论】：

【解决方案2】：

试试这个：

import matplotlib.pyplot as plt
_, ax = plt.subplots(1)
ax.imshow(grid)

【讨论】：