colormapped histogram

jdh2358 · jdh2358 · commit cf3067754549 · 2007-12-04T03:18:39.000Z
svn path=/trunk/matplotlib/; revision=4572
diff --git a/examples/hist_colormapped.py b/examples/hist_colormapped.py
@@ -0,0 +1,24 @@
+import numpy as n
+from pylab import figure, show
+import matplotlib.cm as cm
+import matplotlib.colors as colors
+
+fig = figure()
+ax = fig.add_subplot(111)
+Ntotal = 1000
+N, bins, patches = ax.hist(n.random.rand(Ntotal), 20)
+
+#I'll color code by height, but you could use any scalar
+
+
+# we need to normalize the data to 0..1 for the full
+# range of the colormap
+fracs = N.astype(float)/N.max()
+norm = colors.normalize(fracs.min(), fracs.max())
+
+for thisfrac, thispatch in zip(fracs, patches):
+    color = cm.jet(norm(thisfrac))
+    thispatch.set_facecolor(color)
+
+
+show()
