Update dispersion.py

Fix order of words in y-axis of dispersion_plot
nltk · May 4, 2023 · d7f5e86 · d7f5e86
1 parent 56bc4af
commit d7f5e86
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/nltk/draw/dispersion.py b/nltk/draw/dispersion.py
@@ -44,6 +44,8 @@ def dispersion_plot(text, words, ignore_case=False, title="Lexical Dispersion Pl
             xs.append(x)
             ys.append(y)
 
+    words = list(word2y.keys())
+
     _, ax = plt.subplots()
     ax.plot(xs, ys, "|")
     ax.set_yticks(list(range(len(words))), words, color="C0")