工具 spyder(python3.7)  matplotlib库

在进行路径规划仿真的时候,我们希望最后得到的结果不仅仅是一个 填满数字的数组,而是将它变为更加直观的图片

python路径r python路径规划_python路径r

(spyder数组自带染色,很赞)这是我的A*算法得到的最后的结果,数字意义如下:

数字

0

1

3

4

6

含义

障碍物

可以通行

被加入openlist的节点

被加入closelist节点

最后的路径

但是这样的图片表达效果不佳,我寻求了几种可以将它转化为图片的方法

1.利用pylab中的热图,直接将数组作为参数传入即可得到以下图片

python路径r python路径规划_二维数组_02

代码:

plt.imshow(map_grid, cmap=plt.cm.hot, interpolation='nearest', vmin=0, vmax=10)
plt.colorbar()
xlim(-1, 20)  # 设置x轴范围
ylim(-1, 20)  # 设置y轴范围
my_x_ticks = np.arange(0, 20, 1)
my_y_ticks = np.arange(0, 20, 1)
plt.xticks(my_x_ticks)
plt.yticks(my_y_ticks)
plt.grid(True)
plt.show()

这种方式在表达效果上优于未染色的数组,但是我没有对imshow的热图(heatmap)进行深入研究,所以该没有研究该如何自定义各个部分的颜色,感兴趣的朋友推荐这篇文章给你们

2.直接利用matplotlib进行绘图,利用 plot  scatter以及线条控制来表达我们想要的图像

直接上效果图

python路径r python路径规划_python路径r_03

黑色表示障碍物,浅蓝色方块表示加入过openlist的位置,绿色表示加入closelist的位置,蓝色线条表示我们最后的路径

(是不是感觉好看很多哈哈哈哈)

代码如下:

def draw_path(map_grid,obstacle,path,closelist,openlist=None):
'''
map_grid为处理好的二维数组
obstacle到openlist都是整数,为其对应的标志。比如0代表obstacle,那么obstacle位置就填0
'''
    obstacle_x=[]
    obstacle_y=[]
    path_x=[]
    path_y=[]
    close_list_x=[]
    close_list_y=[]
    open_list_x=[]
    open_list_y=[]
    for i in range(map_grid.shape[0]):
        for j in range(map_grid.shape[1]):
            if map_grid[i][j]==obstacle:   #栅格地图上obstacle为障碍物标识
                obstacle_x.append(i)
                obstacle_y.append(j)    
            if map_grid[i][j]==path:   #栅格地图上path为最后搜索得到路径位置标志
                path_x.append(i)
                path_y.append(j)
            if map_grid[i][j]==closelist:   #栅格地图上closelist为为闭列表中记录的位置标志
                close_list_x.append(i)
                close_list_y.append(j)
            if map_grid[i][j]==openlist:   #栅格地图上openlist为为闭列表中记录的位置标志
                open_list_x.append(i)
                open_list_y.append(j)
    plt.figure(figsize=(10,10))  #为了防止x,y轴间隔不一样长,影响最后的表现效果,所以手动设定等长
    plt.xlim(-1,map_grid.shape[0])
    plt.ylim(-1,map_grid.shape[1])
    my_x_ticks = np.arange(0, map_grid.shape[0], 1)
    my_y_ticks = np.arange(0, map_grid.shape[1], 1)
    plt.xticks(my_x_ticks)#我理解为竖线的位置与间隔
    plt.yticks(my_y_ticks)
    plt.grid(True)  #开启栅格
    plt.plot(path_x,path_y,linewidth=3)
    plt.scatter(open_list_x,open_list_y,s=500,c='cyan',marker='s')
    plt.scatter(obstacle_x,obstacle_y,s=500,c='k',marker='s')
    plt.scatter(close_list_x,close_list_y,s=500,c='g',marker='s')
    plt.title("grid map simulation ")
    plt.show()

简单来讲就是遍历整个地图,然后将需要绘制的障碍物,路径等都通过列表记录下来,最后通过这几行表现出来,但是传入的mapgrid必须是类似本文第一张图那种经过算法处理过的形式

plt.plot(path_x,path_y,linewidth=3)
    plt.scatter(open_list_x,open_list_y,s=500,c='cyan',marker='s')
    plt.scatter(obstacle_x,obstacle_y,s=500,c='k',marker='s')
    plt.scatter(close_list_x,close_list_y,s=500,c='g',marker='s')

marker代表点标志,c表示颜色,s表示标志的大小,s的大小需要大家自己手动调整一下,不一定适用。这里有一份速查博客,可以帮你快速选择自己要的颜色和样式 

说明:这里选择plot来绘制路径是因为plot会自动连线(除非设置linestyle=''),而scatter是绘制散点图,它不会自动连线。

3.利用OpenCV库进行绘制

这种方法理论上可行(我没有实践)

直接通过img=np.zeros(size,dtype='uint8')来创建画布,然后将得到的二维数组做一个比例变换处理(你不希望自己的图只有几百个像素点吧),依靠OpenCV提供的绘图函数来展示内容

这种方法很麻烦,但是对像素点操作的方式决定了它自由度极高,可以满足很多怪异需求

 

这里还是推荐大家使用第二种方式,原理简单,在具有一定自由度的情况下也不需要很麻烦操作。