Do not plot zero-width chords (#17)

Co-authored-by: Tanguy Fardet <Silmathoron@users.noreply.github.com>
tfardet · Nov 25, 2020 · 84d5610 · 84d5610
1 parent 1b211cd
commit 84d5610
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/chord_diagram.py b/chord_diagram.py
@@ -193,9 +193,10 @@ def chord_diagram(mat, names=None, order=None, width=0.1, pad=2., gap=0.03,
 
         start, end = pos[(i, i)]
 
-        self_chord_arc(start, end, radius=1 - width - gap,
-                       chordwidth=0.7*chordwidth, color=colors[i],
-                       alpha=alpha, ax=ax)
+        if mat[i, i] > 0:
+            self_chord_arc(start, end, radius=1 - width - gap,
+                           chordwidth=0.7*chordwidth, color=colors[i],
+                           alpha=alpha, ax=ax)
 
         color = colors[i]
 
@@ -204,10 +205,10 @@ def chord_diagram(mat, names=None, order=None, width=0.1, pad=2., gap=0.03,
 
             start1, end1 = pos[(i, j)]
             start2, end2 = pos[(j, i)]
-
-            chord_arc(start1, end1, start2, end2, radius=1 - width - gap,
-                      chordwidth=chordwidth, color=colors[i], cend=cend,
-                      alpha=alpha, ax=ax, use_gradient=use_gradient)
+            if mat[i, j] > 0 or mat[j, i] > 0:
+                chord_arc(start1, end1, start2, end2, radius=1 - width - gap,
+                          chordwidth=chordwidth, color=colors[i], cend=cend,
+                          alpha=alpha, ax=ax, use_gradient=use_gradient)
 
     # add names if necessary
     if names is not None: