Wordclouds

[This article was first published on R | All Your Bayes, and kindly contributed to R-bloggers]. (You can report issue about the content on this page here)
Want to share your content on R-bloggers? click here if you have a blog, or here if you don't.

TLDR

Wordclouds can be used to produce a neat summary of text and can readily be produced in R. This is a simple example based on a recent conferene paper.


Summarising the content of a conference paper

There is an R package dedicated to creating wordclouds, so I’ve started by loading this, along with the tidyverse (for standard data manipulation) and tidytext (for some help processing the contents of the paper).

library(wordcloud); library(tidyverse); library(tidytext)

The wordcloud package creates a graphic of words that appear in some specified text. The size of the word is proprtional to its frequency in the text
R can read text from a local file, as shown below, or from a website.

# We can read a text file using 'readLines' and we can select a file interactively using 'file.choose'
# Both of these are Base R functions
paper <- readLines(file.choose())

The ‘paper’ variable is currently as list of individual lines, as we can see when viewing one of its elements:

print(paper[2])
## [1] "Application of MCMC Sampling to Account for Variability and Dependency"

tidytext helps get this into a friendlier format allowing us to count the occirence of each word.

paper_tbl <- as_tibble(paper) %>% 
  tidytext::unnest_tokens(word, value) %>% 
  dplyr::filter(is.na(as.numeric(word))) %>% 
  count(word)

Since I expected words like ‘the’ and ‘of’ are likely to feature a lot in the text, I wanted to be able to remove them. I initally used dplyr to set up a variable that would allow me to filter out shorter words, based on some threshold…

minLength <- 4

paper_tbl <- paper_tbl %>% 
  mutate(check = case_when(nchar(word) < minLength ~ 0,
                           nchar(word) >= minLength ~ 1))

But then I learnt about stopwords and made use of the database that tidytext conveniently provides, before removing them from the data.

paper_tbl <- paper_tbl %>%  
 anti_join(tidytext::get_stopwords(language = 'en', source = 'stopwords-iso'))

Before sending this directly into the wordcloud function, we can review the current state of the data, either as a table…

head(x = paper_tbl %>% 
       arrange(desc(x = n)), n = 10)
## # A tibble: 10 x 2
##    word           n
##    <chr>      <int>
##  1 model         58
##  2 models        27
##  3 fatigue       25
##  4 data          24
##  5 parameters    24
##  6 bayesian      22
##  7 posterior     18
##  8 crack         17
##  9 growth        14
## 10 priors        12

…or as a simple plot (in either case I’m only interested in the most frequent words for now) …

ggplot(paper_tbl %>% 
         arrange(desc(n)) %>% 
         dplyr::filter(n >= 12))+
  geom_col(mapping = aes(x = word, y = n))+
  theme_minimal()+ theme(axis.text.x = element_text(angle = 90), 
                         axis.title.x = element_blank())+
  labs(y = 'count')

One thing that is apparent from the above summaries is that we have not dealt with plurals from the data, i.e. ‘model’ and ‘models’ will be treated as two different words, with their own count. I’ve not found a neat way to combine these, but a manual solution with regular expressions (such as grepl, !grepl, etc.) would be simple, though not very elegant. I decided to leave plurals as they are.

Finally, time to ask the wordcloud function to read and plot our data. There are some useful arguments to experiment with here:

  • min.freq and max.words set boundaries for how populated the wordcloud will be
  • random.order will put the largest word in the middle if set to FALSE
  • rot.per is the fraction of words that will be rotated in the graphic

Finally, the words are arranged stochastically somehow, and so for a repeatable graphic we need to specify a seed value.

set.seed(1008) 

wordcloud(words = paper_tbl$word, freq = paper_tbl$n, 
          min.freq = 4, max.words = 100, random.order = FALSE, rot.per = 0.25,
          colors = brewer.pal(n = 8, name = 'Paired'))

If you’re not familiar with the colour palettes, the below line will ask R to display them for you:

RColorBrewer::display.brewer.all()

Finally, some links to more information regarding the packages introduced here, both of which are available on CRAN:

To leave a comment for the author, please follow the link and comment on their blog: R | All Your Bayes.

R-bloggers.com offers daily e-mail updates about R news and tutorials about learning R and many other topics. Click here if you're looking to post or find an R/data-science job.
Want to share your content on R-bloggers? click here if you have a blog, or here if you don't.

Never miss an update!
Subscribe to R-bloggers to receive
e-mails with the latest R posts.
(You will not see this message again.)

Click here to close (This popup will not appear again)