Mercurial Hosting > traffic-intelligence
diff python/utils.py @ 115:550556378466
added functionalities to indicator maps
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Sat, 23 Jul 2011 02:18:16 -0400 |
parents | f03ec4697a09 |
children | 2bf5b76320c0 |
line wrap: on
line diff
--- a/python/utils.py Mon Jul 18 19:24:06 2011 -0400 +++ b/python/utils.py Sat Jul 23 02:18:16 2011 -0400 @@ -174,7 +174,7 @@ colors = PlottingPropertyValues('brgmyck') # 'w' -def plotIndicatorMap(indicatorMap, squareSize): +def plotIndicatorMap(indicatorMap, squareSize, masked = True, defaultValue=-1): from numpy import array, arange, ones, ma from matplotlib.pyplot import pcolor coords = array(indicatorMap.keys()) @@ -182,11 +182,13 @@ minY = min(coords[:,1]) X = arange(minX, max(coords[:,0])+1.1)*squareSize Y = arange(minY, max(coords[:,1])+1.1)*squareSize - C = -ones((len(Y), len(X))) + C = defaultValue*ones((len(Y), len(X))) for k,v in indicatorMap.iteritems(): C[k[1]-minY,k[0]-minX] = v - masked = ma.masked_where(C<0,C) - pcolor(X, Y, masked) + if masked: + pcolor(X, Y, ma.masked_where(C==defaultValue,C)) + else: + pcolor(X, Y, C) ######################### # file I/O section