3 d@sBdZdddZedkr>ddlmZdd d d gZeejd ed S)z. A utility for displaying lexical dispersion. FLexical Dispersion Plotc syddlm}Wn,tk r<}ztd|WYdd}~XnXt|}|j|rttttj|tttj|n||fddt t D}|rtt |\}}nf}}|j ||ddd |j tt t ||d d |jdt ||j||jd |jdS)a$ Generate a lexical dispersion plot. :param text: The source text :type text: list(str) or enum(str) :param words: The target words :type words: list of str :param ignore_case: flag to set if case should be ignored when searching text :type ignore_case: bool )pylabzQThe plot function requires matplotlib to be installed.See https://matplotlib.org/Ncs6g|].}ttD]}||kr||fqqS)rangelen).0xy) text_to_comp words_to_compr4/tmp/pip-build-v9q4h5k9/nltk/nltk/draw/dispersion.py ,sz#dispersion_plot..zb|g?)Zscalexb)colorz Word Offset)Z matplotlibr ImportError ValueErrorlistreversemapstrlowerrrzipZplotZyticksZylimtitleZxlabelshow) textwordsZ ignore_caserreZpointsrr r)r r r dispersion_plot s0    r__main__r) gutenbergZElinorZMarianneZEdwardZ Willoughbyzausten-sense.txtN)Fr)__doc__r__name__Z nltk.corpusr!rrrrr  s  0