partial coloring of text: unicorn poop example

Author: ivanov

examples/text_labels_and_annotations/rainbow_text.py

@@ -0,0 +1,53 @@
+# -*- coding: utf-8 -*-
+"""
+On the matplotlib-users list back in February 2012, Gökhan Sever asked the
+following question:
+
+    Is there a way in matplotlib to partially specify the color of a string?
+
+    Example:
+
+    plt.ylabel("Today is cloudy.")
+    How can I show "today" as red, "is" as green and "cloudy." as blue?
+
+    Thanks.
+
+Paul Ivanov responded with this answer:
+"""
+
+import matplotlib.pyplot as plt
+from matplotlib import transforms
+
+def rainbow_text(x,y,ls,lc,**kw):
+    """
+    Take a list of strings ``ls`` and colors ``lc`` and place them next to each
+    other, with text ls[i] being shown in color lc[i].
+
+    This example shows how to do both vertical and horizontal text, and will
+    pass all keyword arguments to plt.text, so you can set the font size,
+    family, etc.
+    """
+    t = plt.gca().transData
+    fig = plt.gcf()
+    plt.show()
+
+    #horizontal version
+    for s,c in zip(ls,lc):
+        text = plt.text(x, y, " " + s + " ", color=c, transform=t, **kw)
+        text.draw(fig.canvas.get_renderer())
+        ex = text.get_window_extent()
+        t = transforms.offset_copy(text._transform, x=ex.width, units='dots')
+
+    #vertical version
+    for s,c in zip(ls,lc):
+        text = plt.text(x, y, " " + s + " ", color=c, transform=t,
+                rotation=90, va='bottom', ha='center', **kw)
+        text.draw(fig.canvas.get_renderer())
+        ex = text.get_window_extent()
+        t = transforms.offset_copy(text._transform, y=ex.height, units='dots')
+
+
+plt.figure()
+rainbow_text(0.0,0.0,"all unicorns poop rainbows ! ! !".split(), 
+        ['red', 'orange', 'brown', 'green', 'blue', 'purple', 'black'],
+        size=30)