Changeset 609

Timestamp:

09/02/16 00:06:41 (8 years ago)

Author:

Maciej Komosinski

Message:

Labels can have colors dependent on label text

File:

• mds-and-trees/mds_plot.py (modified) (1 diff)

mds-and-trees/mds_plot.py

@@ -118 +118 @@
         points = [add_jitter(coordinates[:, i]) for i in range(x_dim)]
         
-        if labels is not None and dimensions==2:
-                ax.scatter(*points, alpha=0.1) #barely visible points, because we will show labels anyway
+        if labels is not None and dimensions==2: #could be ported to 3D too
+                ax.scatter(*points, alpha=0) #invisible points, because we will show labels instead
                 labelconvert={'velland':'V','velwat':'W','vpp':'P','vpa':'A'} #use this if you want to replace long names with short IDs
+                colors={'velland':'green','velwat':'blue','vpp':'red','vpa':'violet'}
                 #for point in points:
                 #       print(point)
                 for label, x, y in zip(labels, points[0], points[1]):
+                        color='black'
                         for key in labelconvert:
                                 if label.startswith(key):
                                         label=labelconvert[key]
+                                        color=colors[key]
                         plt.annotate(
                                 label, 
                                 xy = (x, y), xytext = (0, 0),
                                 textcoords = 'offset points', ha = 'center', va = 'center',
+                                color = color,
+                                alpha = 0.8,
                                 #bbox = dict(boxstyle = 'round,pad=0.5', fc = 'yellow', alpha = 0.5),
                                 #arrowprops = dict(arrowstyle = '->', connectionstyle = 'arc3,rad=0')