Mercurial > hg > nsaunier > traffic-intelligence

comparison python/utils.py @ 940:d8ab183a7351

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
verified motion prediction with prototypes at constant speed (test needed)
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Tue, 18 Jul 2017 13:46:17 -0400
parents c7e72d758049
children 5d788d2e8ffc
comparison
equal deleted inserted replaced
939:a2f3f3ca241e 940:d8ab183a7351
834 834
835 ######################### 835 #########################
836 # plotting section 836 # plotting section
837 ######################### 837 #########################
838 838
839 def plotPolygon(poly, options = ''): 839 def plotPolygon(poly, options = '', **kwargs):
840 'Plots shapely polygon poly' 840 'Plots shapely polygon poly'
841 from numpy.core.multiarray import array
842 from matplotlib.pyplot import plot 841 from matplotlib.pyplot import plot
843 from shapely.geometry import Polygon 842 x,y = poly.exterior.xy
844 843 plot(x, y, options, **kwargs)
845 tmp = array(poly.exterior)
846 plot(tmp[:,0], tmp[:,1], options)
847 844
848 def stepPlot(X, firstX, lastX, initialCount = 0, increment = 1): 845 def stepPlot(X, firstX, lastX, initialCount = 0, increment = 1):
849 '''for each value in X, increment by increment the initial count 846 '''for each value in X, increment by increment the initial count
850 returns the lists that can be plotted 847 returns the lists that can be plotted
851 to obtain a step plot increasing by one for each value in x, from first to last value 848 to obtain a step plot increasing by one for each value in x, from first to last value