Statistical Computing, 36-350
Monday September 16, 2019
subset()
: function for extracting rows of a data frame meeting a conditionsplit()
: function for splitting up rows of a data frame, according to a factor variableapply()
: function for applying a given routine to rows or columns of a matrix or data framelapply()
: similar, but used for applying a routine to elements of a vector or listsapply()
: similar, but will try to simplify the return type, in comparison to lapply()
tapply()
: function for applying a given routine to groups of elements in a vector or list, according to a factor variableString basics
The simplest distinction:
Character: a symbol in a written language, like letters, numerals, punctuation, space, etc.
String: a sequence of characters bound together
## [1] "character"
## [1] "character"
Why do we care about strings?
Whitespaces count as characters and can be included in strings:
" "
for space"\n"
for newline"\t"
for tab## [1] "Dear Mr. Carnegie,\n\nThanks for the great school!\n\nSincerely, Ryan"
Use cat()
to print strings to the console, displaying whitespaces properly
## Dear Mr. Carnegie,
##
## Thanks for the great school!
##
## Sincerely, Ryan
The character is a basic data type in R (like numeric, or logical), so we can make vectors or matrices of out them. Just like we would with numbers
str.vec = c("Statistical", "Computing", "isn't that bad") # Collect 3 strings
str.vec # All elements of the vector
## [1] "Statistical" "Computing" "isn't that bad"
## [1] "isn't that bad"
## [1] "isn't that bad"
str.mat = matrix("", 2, 3) # Build an empty 2 x 3 matrix
str.mat[1,] = str.vec # Fill the 1st row with str.vec
str.mat[2,1:2] = str.vec[1:2] # Fill the 2nd row, only entries 1 and 2, with
# those of str.vec
str.mat[2,3] = "isn't a fad" # Fill the 2nd row, 3rd entry, with a new string
str.mat # All elements of the matrix
## [,1] [,2] [,3]
## [1,] "Statistical" "Computing" "isn't that bad"
## [2,] "Statistical" "Computing" "isn't a fad"
## [,1] [,2]
## [1,] "Statistical" "Statistical"
## [2,] "Computing" "Computing"
## [3,] "isn't that bad" "isn't a fad"
Easy! Make things into strings with as.character()
## [1] "0.8"
## [1] "8e+09"
## [1] "1" "2" "3" "4" "5"
## [1] "TRUE"
Not as easy! Depends on the given string, of course
## [1] 0.5
## [1] 0.5
## [1] 5e-11
## Warning: NAs introduced by coercion
## [1] NA
## [1] TRUE
## [1] NA
Use nchar()
to count the number of characters in a string
## [1] 6
## [1] 11
## [1] 1
## [1] 2
## [1] 6 11
Substrings, splitting and combining strings
Use substr()
to grab a subseqence of characters from a string, called a substring
## [1] "Give"
## [1] "break"
## [1] ""
substr()
vectorizesJust like nchar()
, and many other string functions
presidents = c("Clinton", "Bush", "Reagan", "Carter", "Ford")
substr(presidents, 1, 2) # Grab the first 2 letters from each
## [1] "Cl" "Bu" "Re" "Ca" "Fo"
## [1] "C" "u" "a" "t" ""
## [1] "C" "Bu" "Rea" "Cart" "Ford"
## [1] "on" "sh" "an" "er" "rd"
Can also use substr()
to replace a character, or a substring
## [1] "Give me a break"
## [1] "Live me a break"
## [1] "Live me a break"
## [1] "Show me a break"
Use the strsplit()
function to split based on a keyword
ingredients = "chickpeas, tahini, olive oil, garlic, salt"
split.obj = strsplit(ingredients, split=",")
split.obj
## [[1]]
## [1] "chickpeas" " tahini" " olive oil" " garlic" " salt"
## [1] "list"
## [1] 1
Note that the output is actually a list! (With just one element, which is a vector of strings)
strsplit()
vectorizesJust like nchar()
, substr()
, and the many others
great.profs = "Nugent, Genovese, Greenhouse, Seltman, Shalizi, Ventura"
favorite.cats = "tiger, leopard, jaguar, lion"
split.list = strsplit(c(ingredients, great.profs, favorite.cats), split=",")
split.list
## [[1]]
## [1] "chickpeas" " tahini" " olive oil" " garlic" " salt"
##
## [[2]]
## [1] "Nugent" " Genovese" " Greenhouse" " Seltman" " Shalizi"
## [6] " Ventura"
##
## [[3]]
## [1] "tiger" " leopard" " jaguar" " lion"
strsplit()
needs to return a list now?Finest splitting you can do is character-by-character: use strsplit()
with split=""
## [1] "c" "h" "i" "c" "k" "p" "e" "a" "s" "," " " "t" "a" "h" "i" "n" "i"
## [18] "," " " "o" "l" "i" "v" "e" " " "o" "i" "l" "," " " "g" "a" "r" "l"
## [35] "i" "c" "," " " "s" "a" "l" "t"
## [1] 42
## [1] 42
Use the paste()
function to join two (or more) strings into one, separated by a keyword
## [1] "Spider Man"
## [1] "Spider-Man"
## [1] "Spider, Man, does whatever"
paste()
vectorizesJust like nchar()
, substr()
, strsplit()
, etc. Seeing a theme yet?
## [1] "Clinton" "Bush" "Reagan" "Carter" "Ford"
## [1] "Clinton D" "Bush R" "Reagan R" "Carter D" "Ford R"
## [1] "Clinton D" "Bush R" "Reagan D" "Carter R" "Ford D"
## [1] "Clinton (42)" "Bush (41)" "Reagan (40)" "Carter (39)"
## [5] "Ford (38)"
Can condense a vector of strings into one big string by using paste()
with the collapse
argument
## [1] "Clinton" "Bush" "Reagan" "Carter" "Ford"
## [1] "Clinton; Bush; Reagan; Carter; Ford"
## [1] "Clinton (42); Bush (41); Reagan (40); Carter (39); Ford (38)"
## [1] "Clinton (D42); Bush (R41); Reagan (R40); Carter (D39); Ford (R38)"
## [1] "Clinton" "Bush" "Reagan" "Carter" "Ford"
Reading in text, summarizing text
How to get text, from an external source, into R? Use the readLines()
function
trump.lines = readLines("http://www.stat.cmu.edu/~ryantibs/statcomp-F19/data/trump.txt")
class(trump.lines) # We have a character vector
## [1] "character"
## [1] 113
## [1] "Friends, delegates and fellow Americans: I humbly and gratefully accept your nomination for the presidency of the United States."
## [2] "Story Continued Below"
## [3] ""
(This was Trump’s acceptance speech at the 2016 Republican National Convention)
We don’t need to use the web; readLines()
can be used on a local file. The following code would read in a text file from Professor Tibs’ computer:
## Warning in file(con, "r"): cannot open file '/Users/ryantibshirani/Dropbox/
## teaching/f19-350/lectures/text/trump.txt': No such file or directory
## Error in file(con, "r"): cannot open the connection
This will cause an error for you, unless your folder is set up exactly like Professor Tibs’ laptop! So using web links is more robust
Fancy word, but all it means: make one long string, then split the words
trump.text = paste(trump.lines, collapse=" ")
trump.words = strsplit(trump.text, split=" ")[[1]]
# Sanity check
substr(trump.text, 1, 200)
## [1] "Friends, delegates and fellow Americans: I humbly and gratefully accept your nomination for the presidency of the United States. Story Continued Below Together, we will lead our party back to the Whi"
## [1] "Friends," "delegates" "and" "fellow" "Americans:"
## [6] "I" "humbly" "and" "gratefully" "accept"
## [11] "your" "nomination" "for" "the" "presidency"
## [16] "of" "the" "United" "States." "Story"
Our most basic tool for summarizing text: word counts, retrieved using table()
## [1] "table"
## [1] 1604
## trump.words
## – 'I "extremely "I’m
## 1 34 1 1 1
## "I’M "negligent," $150 $19 $2
## 1 1 1 1 1
What did we get? Alphabetically sorted unique words, and their counts = number of appearances
Note: this is actually a vector of numbers, and the words are the names of the vector
## trump.words
## – 'I "extremely "I’m
## 1 34 1 1 1
## –
## TRUE
## [1] TRUE
So with named indexing, we can now use this to look up whatever words we want
## America
## 19
## great
## 7
## wall
## 1
## <NA>
## NA
Let’s sort in decreasing order, to get the most frequent words
## [1] 1604
## trump.words
## the and of to our will in I have a that for
## 189 145 127 126 90 82 69 64 57 51 48 46
## is are we – their be on was
## 40 39 35 34 28 26 26 26
## trump.words
## wonder workers workforce works, worth wouldn’t
## 1 1 1 1 1 1
## wounded years-old, years. yet Yet Yet,
## 1 1 1 1 1 1
## YOU you, You, you: YOU. youngest
## 1 1 1 1 1 1
## YOUR youth
## 1 1
Notice that punctuation matters, e.g., “Yet” and “Yet,” are treated as separate words, not ideal—we’ll learn just a little bit about how to fix this on lab/homework, using regular expressions
Let’s use a plot to visualize frequencies
nw = length(trump.wordtab.sorted)
plot(1:nw, as.numeric(trump.wordtab.sorted), type="l",
xlab="Rank", ylab="Frequency")
A pretty drastic looking trend! It looks as if \(\mathrm{Frequency} \propto (1/\mathrm{Rank})^a\) for some \(a>0\)
This phenomenon, that frequency tends to be inversely proportional to a power of rank, is called Zipf’s law
For our data, Zipf’s law approximately holds, with \(\mathrm{Frequency} \approx C(1/\mathrm{Rank})^a\) for \(C=215\) and \(a=0.57\)
C = 215; a = 0.57
trump.wordtab.zipf = C*(1/1:nw)^a
cbind(trump.wordtab.sorted[1:8], trump.wordtab.zipf[1:8])
## [,1] [,2]
## the 189 215.00000
## and 145 144.82761
## of 127 114.94216
## to 126 97.55831
## our 90 85.90641
## will 82 77.42697
## in 69 70.91410
## I 64 65.71691
Not perfect, but not bad. We can also plot the original sorted word counts, and those estimated by our formula law on top
plot(1:nw, as.numeric(trump.wordtab.sorted), type="l",
xlab="Rank", ylab="Frequency")
curve(C*(1/x)^a, from=1, to=nw, col="red", add=TRUE)
We’ll learn about plotting tools in detail a bit later
nchar()
, substr()
: functions for substring extractions and replacementsstrsplit()
, paste()
: functions for splitting and combining stringstable()
: function to get word counts, useful way of summarizing text data