python 快速实现栅格地图

依赖库：

import matplotlib.pyplot as plt
import numpy as np
from matplotlib import colors

实现过程：使用numpy生成所需栅格地图的矩阵值。在此基础上随机生成障碍，值为1代表障碍，值为0代表无障碍，最后使用matplotlib绘画出。

代码如下：

 # 地图长，宽
 map_w = 10
 map_h =10

 G = np.zeros(map_w*map_h)

 obs_num = int(G.size * 0.2)  # 障碍数量
 obs_a = np.random.randint(low=0, high=G.size, size=obs_num)  # 随机产生障碍的索引
 G[obs_a] = 1  # 障碍
 map_test = G.reshape([map_w, map_h])  # 创建矩阵地图

cmap = colors.ListedColormap(['none', 'black', 'white', 'magenta', 'yellow', 'cyan', 'green', 'red', 'blue'])
plt.imshow(map_test, cmap=cmap, interpolation='nearest', vmin=0, vmax=7)
plt.show()

最后效果图：