Merge pull request #18 from selimelawwa/add_world_clouds

Initial commit to support world-clouds visualization

requirements.txt

@@ -2,3 +2,4 @@ nltk
 scikit-learn
 plotly_express
 spacy
+wordcloud

texthero/visualization.py

@@ -4,7 +4,7 @@
 
 import pandas as pd
 import plotly.express as px
-
+from wordcloud import WordCloud
 
 def scatterplot(df: pd.DataFrame,
                 col: str,
@@ -38,6 +38,29 @@ def scatterplot(df: pd.DataFrame,
     if return_figure:
         return fig
 
+def wordcloud(s: pd.Series, 
+              title="",
+              return_figure=False):
+    """
+    Show wordcloud using WordCloud.
+
+    Parameters
+    ----------
+    df
+    col
+        The name of the column of the DataFrame containing the text data.
+
+    """
+    text = s.str.cat(sep=' ')
+
+    wordcloud = WordCloud(background_color ='white', min_font_size = 10).generate(text) 
+
+    fig = px.imshow(wordcloud, title=title)
+    fig.show()
+
+    if return_figure:
+        return fig
+
 
 def top_words(s: pd.Series, normalize=False) -> pd.Series:
     """