In Sankey, prevented appended lists of labels

kdavies4 · kdavies4 · commit f813278c26d7 · 2012-12-07T08:03:11.000-05:00
diff --git a/lib/matplotlib/sankey.py b/lib/matplotlib/sankey.py
@@ -755,22 +755,18 @@ def _get_angle(a, r):
         self.ax.add_patch(patch)
 
         # Add the path labels.
-        for i, (number, angle) in enumerate(zip(flows, angles)):
-            if labels[i] is None or angle is None:
-                labels[i] = ''
+        texts = []
+        for i, (number, angle, label, location) in enumerate(zip(flows, angles,
+                                                     labels, label_locations)):
+            if label is None or angle is None:
+                label = ''
             elif self.unit is not None:
                 quantity = self.format % abs(number) + self.unit
-                if labels[i] != '':
-                    labels[i] += "\n"
-                labels[i] += quantity
-        texts = []
-        for i, (label, location) in enumerate(zip(labels, label_locations)):
-            if label:
-                s = label
-            else:
-                s = ''
+                if label != '':
+                    label += "\n"
+                label += quantity
             texts.append(self.ax.text(x=location[0], y=location[1],
-                                      s=s,
+                                      s=label,
                                       ha='center', va='center'))
         # Text objects are placed even they are empty (as long as the magnitude
         # of the corresponding flow is larger than the tolerance) in case the
