image_to_laser_moves.py 文件源码

def plotResults(result):
    from pylab import figure, hold, show
    from matplotlib import collections  as mc
    figure()
    hold(True)
    lines = []
    mWs = []
    for seg_i in range(1, len(result)):
        lines.append([result[seg_i-1][1:3], result[seg_i][1:3]])
        mWs.append(result[seg_i][3])

    norm = mpl.colors.Normalize(vmin=0, vmax=150)
    colorMapper = cm.ScalarMappable(norm=norm, cmap=cm.jet)
    lc = mc.LineCollection(lines, linewidths=2, colors=colorMapper.to_rgba(mWs))
    ax = gca()
    ax.add_collection(lc)
    ax.autoscale()
    ax.margins(0.1)
    ax.axis('equal')

    #plot(result[:,2], result[:,1])
    #scatter(result[:,2], result[:,1], c=result[:,3],vmin=0, vmax=150)
    #colorbar(colorMapper)
    show()