matplotlib-使用Python和numpy打印二维地图数据

我正在尝试绘制CSV格式的地图数据，以便执行机器人定位。我有一个名为

map

的numpy 2D数组，它根据机器人位于网格位置的概率值进行索引

例如，

map[500375]=0.75

表示机器人有75%的几率处于网格

500375

目前，我正在使用matplotlib的

surface\u plot

函数绘制三维高度贴图，但我希望使用matplotlib绘制二维颜色贴图，其中概率由特定颜色表示，而不是高度

---

有什么办法吗？

事实上，我不确定这是否是您需要的，但我想您可能想看看。它有多种颜色贴图

例如，使用

热cmap

绘制某物：

plt.imshow(map, origin='lower', cmap="hot", interpolation='none')

参数实际上并不十分重要，

原点（从我的角度来看）混淆了
0,0
默认为左上角，如果希望
0,0
为左下角，则需要指定
'lower'
。如果需要某种插值，可以省略插值，但如果不需要插值，则需要将其定义为
'none'

可以在其中一个MatPlotLib中找到所有可能的颜色映射的列表