我正在尝试绘制以 CSV 格式给出的 map 数据以执行机器人定位。我有一个名为 map
的 numpy 2D 数组,它索引机器人位于该网格位置的概率值。
例如map[500, 375] = 0.75
表示机器人有 75% 的机会位于网格 500, 375
。
目前我正在使用 matplotlib 的 surface_plot
函数来绘制 3D 高度图,但我希望使用 matplotlib 来绘制 2D 颜色图,其中概率由特定颜色表示,而不是高度。
关于如何做到这一点有什么想法吗?
最佳答案
实际上我不确定这是否是您所需要的,但我想您可能想看看plt.imshow()
。它有各种各样的颜色图。
例如使用 hot-cmap
绘制某些内容:
plt.imshow(map, origin='lower', cmap="hot", interpolation='none')
这些参数实际上并不是很重要,origin
(从我的角度来看)令人困惑,0,0
默认位于左上角,您需要指定'lower'
如果您希望 0,0
位于左下角。如果您想要某种插值,则可以省略插值,但如果您不想要任何插值,则需要将其定义为'none'
。
所有可能的颜色图的列表可以在 matplotlibs example codes 之一中找到。 .
关于python - matplotlib - 使用 Python 和 numpy 绘制 2D map 数据,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/35591429/