Merge pull request nltk#3158 from dlukes/feature/dispersion-plot-xaxis

Make sure dispersion plot spans full text length

Author: stevenbird

AUTHORS.md

@@ -110,6 +110,7 @@
 - Tom Lippincott
 - Peter Ljunglöf
 - Alex Louden
+- David Lukeš
 - Joseph Lynch
 - Nitin Madnani
 - Felipe Madrigal

nltk/draw/dispersion.py

@@ -48,6 +48,8 @@ def dispersion_plot(text, words, ignore_case=False, title="Lexical Dispersion Pl
 
     _, ax = plt.subplots()
     ax.plot(xs, ys, "|")
+    ax.dataLim.x0, ax.dataLim.x1 = 0, len(text) - 1
+    ax.autoscale(axis="x")
     ax.set_yticks(list(range(len(words))), words, color="C0")
     ax.set_ylim(-1, len(words))
     ax.set_title(title)