Check R help for on Control Flow ?Control.
for(){} for(loop control){do something each iteration}
for(iterator in vector){
#do something
}Loop control is defined in between the parentheses. The name of the iterator is placed on the left of in(can be assigned any name you want, does not need to be declared in advance). During the execution of the loop, the iterator takes on the values inside the vector which is placed on the right side of in. Specifically, the following is happening.
Loop steps: 1. iterator <- vector[1] 2. iterator <- vector[2] 3. iterator <- vector[3] 4. etc.
The loop will automatically stop once it reaches the last item in the vector. The loop can be stopped before that using the break command.
# Make a loop do something 5 times
# i is the iterator
# 1:5 creates a vector with 5 numbers in it, 1, 2, 3, 4, 5
# the loop will run 5 times, because there are five things to assign to i
for(i in 1:5){
print(5)
}## [1] 5
## [1] 5
## [1] 5
## [1] 5
## [1] 5# show the value of i each step of the loop
for(i in 1:5){
print(i)
}## [1] 1
## [1] 2
## [1] 3
## [1] 4
## [1] 5# define the vector to loop over in advance
my_sequence <- 1:5
for(i in my_sequence){
print(i)
}## [1] 1
## [1] 2
## [1] 3
## [1] 4
## [1] 5# Reminder that i becomes the next value in the vector
# your vector can have any order
my_sequence <- c(1,5,2,3,4)
for(i in my_sequence){
print(i)
}## [1] 1
## [1] 5
## [1] 2
## [1] 3
## [1] 4# index vector does not need to be numbers
my_things <- c("A","B","C","D")
for(i in my_things){
print(i)
}## [1] "A"
## [1] "B"
## [1] "C"
## [1] "D"break stops a loop. Used with logical statements to define the conditions necessary to cause the break.
for(i in 1:10){
if(i <5){
print(i)
} else{
break
}
}## [1] 1
## [1] 2
## [1] 3
## [1] 4While loops run until a logical condition is met. Here there is no iterator, just a logic statement that needs to be met.
This one prints i while i is less than 6. As soon as i becomes “not less than 6”, then the loop stops. Critically, inside the loop, the value of i increases each iteration.
i <- 1 # create an variable
while (i < 6) {
print(i)
i = i+1 #add one eachs step of the loop
}## [1] 1
## [1] 2
## [1] 3
## [1] 4
## [1] 5Similar to while, but let’s do things until a condition is met.
i<-0
repeat{
i<-i+1
print(i)
if(i==5){
break
}
}## [1] 1
## [1] 2
## [1] 3
## [1] 4
## [1] 5Braces are not needed on one line
for(i in 1:5) print(i)## [1] 1
## [1] 2
## [1] 3
## [1] 4
## [1] 5Using the value of the iterator to assign in values systematically to another variable.
# put 1 into the first five positions of x
x <- c() # create empty vector
for(i in 1:5){
x[i] <- 1 # assign 1 to the ith slot in x
}
x## [1] 1 1 1 1 1# put the numbers 1-5 in the first 5 positions of x
x <-c()
for(i in 1:5){
x[i] <- i
}
x## [1] 1 2 3 4 5Make your own counter, when you need one
a <- c(1,4,3,5,7,6,8,2)
odd <- c()
counter <- 0
for(i in a){ # i will the values of a in each position
counter <- counter+1
if(i%%2 != 0){
odd[counter] <- "odd"
} else {
odd[counter] <- "even"
}
}
odd## [1] "odd" "even" "odd" "odd" "odd" "even" "even" "even"# An alternative strategy
a <- c(1,4,3,5,7,6,8,2)
odd <- c()
# 1:length(a) creates a sequence from 1 to length
for(i in 1:length(a)){
if(a[i]%%2 != 0){
odd[i] <- "odd"
} else {
odd[i] <- "even"
}
}
odd## [1] "odd" "even" "odd" "odd" "odd" "even" "even" "even"Nesting loops
for(i in 1:5){
for(j in 1:5){
print(c(i,j))
}
}## [1] 1 1
## [1] 1 2
## [1] 1 3
## [1] 1 4
## [1] 1 5
## [1] 2 1
## [1] 2 2
## [1] 2 3
## [1] 2 4
## [1] 2 5
## [1] 3 1
## [1] 3 2
## [1] 3 3
## [1] 3 4
## [1] 3 5
## [1] 4 1
## [1] 4 2
## [1] 4 3
## [1] 4 4
## [1] 4 5
## [1] 5 1
## [1] 5 2
## [1] 5 3
## [1] 5 4
## [1] 5 5# example of using nested loops to fill the contents
# of a matrix
my_matrix <- matrix(0,ncol=5,nrow=5)
for(i in 1:5){
for(j in 1:5){
my_matrix[i,j] <- i*j
}
}
my_matrix## [,1] [,2] [,3] [,4] [,5]
## [1,] 1 2 3 4 5
## [2,] 2 4 6 8 10
## [3,] 3 6 9 12 15
## [4,] 4 8 12 16 20
## [5,] 5 10 15 20 25break exits out of the immediate loop
# the inside loop stops when i+j is greater than 5
# the outside loop keeps going
sum_of_i_j <- c()
counter <- 0
for(i in 1:5){
for(j in 1:5){
counter <- counter+1
sum_of_i_j[counter] <- i+j
if(i+j > 5){
break
}
}
}
sum_of_i_j## [1] 2 3 4 5 6 3 4 5 6 4 5 6 5 6 6Logic statements are used to compare two things, or two sets of things. The output of comparison is a TRUE or FALSE statment. If many things are being compared at once, the output could be many TRUE or FALSE statements for each comparison
1==1 # is 1 equal to 1?## [1] TRUE1==2 # is 1 equal to 2?## [1] FALSEc(1,2,3) == c(2,1,3) # compares each element with each element## [1] FALSE FALSE TRUE1 == c(2,1,3)## [1] FALSE TRUE FALSE1!=1 # is 1 equal to 1?## [1] FALSE1!=2 # is 1 equal to 2?## [1] TRUEc(1,2,3) != c(2,1,3) # compares each element with each element## [1] TRUE TRUE FALSE1 != c(2,1,3)## [1] TRUE FALSE TRUE1 > 1 # is 1 greater than 1?## [1] FALSE5 > 1 # is 5 greater than 1?## [1] TRUE3 < 2 # is 3 less than 2?## [1] FALSE3 < 1 # is 3 less than 1?## [1] FALSEc(1,2,3) > c(2,1,3) # ask the question element by element## [1] FALSE TRUE FALSEc(1,2,3) < c(2,1,3)## [1] TRUE FALSE FALSE2 > c(1,2,3) # is greater than each of the numbers## [1] TRUE FALSE FALSEIs something greater than or equal to something else
1 >= 1 # is 1 greater than 1?## [1] TRUE5 >= 1 # is 5 greater than 1?## [1] TRUE3 <= 2 # is 3 less than 2?## [1] FALSE3 <= 1 # is 3 less than 1?## [1] FALSEc(1,2,3) >= c(2,1,3) # ask the question element by element## [1] FALSE TRUE TRUEc(1,2,3) <= c(2,1,3)## [1] TRUE FALSE TRUE2 >= c(1,2,3) # is greater than each of the numbers## [1] TRUE TRUE FALSEThe ampersand & is used for AND, which allows use to evaluate whether two or more properties are all TRUE.
# is 16 divisible by 4 AND 8
16%%4 == 0 & 16%%8 ==0## [1] TRUE# is 16 divisible by 4 AND 3
16%%4 == 0 & 16%%3 ==0## [1] FALSE# is 16 divisible by 8 and 4 and 2
16%%4 == 0 & 16%%8 ==0 & 16%%2 ==0## [1] TRUEThe | is used for OR, which allows use to evaluate at least one of the properties is TRUE.
# is 16 divisible by 4 OR 8
16%%4 == 0 | 16%%8 ==0## [1] TRUE# is 16 divisible by 4 OR 3
# it is divisible by 4, so the answer is TRUE
# because at least one of the comparisons is TRUE
16%%4 == 0 | 16%%3 ==0## [1] TRUEWhen R returns values as TRUE or FALSE, it return a logical variable. It also treats TRUE as a 1, and FALSE as a 0. In the example below we see it is possible sum up a logical variable with multiple TRUE and FALSE entries.
c(1,2,3) == c(1,2,3)## [1] TRUE TRUE TRUEsum(c(1,2,3) == c(1,2,3))## [1] 3c(1,2,3) == c(2,1,3)## [1] FALSE FALSE TRUEsum(c(1,2,3) == c(2,1,3))## [1] 1A carnival operator needs to check if people are taller than a line to see if they can ride the ride. Every time someone goes through the gate, they run an IF ELSE control structure in their head. IF the person is taller than the line, then they can go on the ride; ELSE (otherwise) the person can not go on the ride.
In other words, IF the situation is X, then do something; ELSE (if the situation is not X), then do something different.
IF and ELSE statements let us specify the conditions when specific actions are taken. Generally, IF and ELSE statements are used inside loops (for, or while, or repeat loops), because at each step or iteration of the loop, we want to check something, and then do something.
Consider this:
a <- 1 # define a to be a 1
if(a==1){
print(a) # this is what happens if a==1
} else {
print("A is not 1") # this is what happens if a is not 1
}## [1] 1a <- 2 # define a to be a 1
if(a==1){
print(a) # this is what happens if a==1
} else {
print("A is not 1") # this is what happens if a is not 1
}## [1] "A is not 1"Normally we find IF and ELSE in a loop like this:
a <- c(1,0,1,0,0,0,1) # make a variable contain 1s and 0s
# write a loop to check each element in the variable
# and do different things depending on the element
for(i in a){
if(i == 1){
print("I'm a 1") # what to do when i is 1
} else {
print("I'm not a 1") # what to do when i is not 1
}
}## [1] "I'm a 1"
## [1] "I'm not a 1"
## [1] "I'm a 1"
## [1] "I'm not a 1"
## [1] "I'm not a 1"
## [1] "I'm not a 1"
## [1] "I'm a 1"We can have multiple conditions in our if statements.
a <- c(1,2,3,1,2,0,1) # make a variable contain 1s and 0s
# write a loop to check each element in the variable
# and do different things depending on the element
for(i in a){
if(i == 1){
print("I'm a 1") # what to do when i is 1
} else if (i==2){
print("I'm a 2") # what to do when i is 2
} else if (i==3){
print("I'm a 3") # what to do when i is 3
} else {
print("I'm not any of the above") #what to do when none are true
}
}## [1] "I'm a 1"
## [1] "I'm a 2"
## [1] "I'm a 3"
## [1] "I'm a 1"
## [1] "I'm a 2"
## [1] "I'm not any of the above"
## [1] "I'm a 1"This section discusses the syntax for writing custom functions in R.
function_name <- function(input1,input2){
#code here
return(something)
}This function has no input between the (). Whenever you run this function, it will simply return whatever is placed inside the return statement.
# define the function
print_hello_world <- function(){
return(print("hello world"))
}
# use the function
print_hello_world()## [1] "hello world"This function simply takes an input, and then returns the input without modifying it.
return_input <- function(input){
return(input)
}
# the variable input is assigned a 1
# then we return(input), which will result in a 1
# because the function internally assigns 1 to the input
return_input(1)## [1] 1a <- "something"
return_input(a)## [1] "something"This function takes an input, then creates an internal variable called temp and assigns input+1. Then the contents of temp is returned. Note there, is no checking of the input, so it will return an erro if you input a character (can’t add one to a character in R)
add_one <- function(input){
temp <- input+1
return(temp)
}
add_one(1)## [1] 2add_one("a")## Error in input + 1: non-numeric argument to binary operatorThis function adds some input checking. We only add one if the input is a numeric type. Otheriwse, we use stop() to return an error message to the console
add_one <- function(input){
if(class(input) == "numeric"){
temp <- input+1
return(temp)
} else {
return(stop("input must be numeric"))
}
}
add_one(1)## [1] 2add_one("a")## Error in add_one("a"): input must be numericA function with three inputs
add_multiply <- function(input, x_plus,x_times){
temp <- (input+x_plus)*x_times
return(temp)
}
# input is 1
# x_plus <- 2
# x_times <- 3
# will return (1+2)*3 = 9
add_multiply(1,2,3)## [1] 9Loops are a common tool for doing something many times. R can accomplish the goal of “doing something many times” without loops, using a vectorized approach.
Let’s take a close look at some very basic differences between using a loop, and using R’s vectorized approach
Consider the problem of adding a single number to all of the numbers in a vector.
nums <- c(1,2,3,4)
# vectorized approach
# R automatically adds 1 to all of the numbers
nums+1## [1] 2 3 4 5# loop approach
# much longer to write out
for(i in 1:length(nums)){
nums[i] <- nums[i]+1
}
nums## [1] 2 3 4 5How about adding two vectors together, so we add the first two numbers together, then the second two numbers etc.
A <- c(1,2,3,4)
B <- c(1,2,3,4)
# vectorized approach
A+B## [1] 2 4 6 8# loop approach
the_sum <-c()
for(i in 1:length(A)){
the_sum[i] <- A[i]+B[i]
}
the_sum## [1] 2 4 6 8How about comparing the identity of the elements in two vectors to see if they are the same or not?
A <- c("c","e","f","g")
B <- c("d","e","f","g")
#vectorized approach
A==B## [1] FALSE TRUE TRUE TRUE# loop approach
compared <-c()
for(i in 1:length(A)){
if(A[i]==B[i]){
compared[i] <- TRUE
} else {
compared[i] <- FALSE
}
}
compared## [1] FALSE TRUE TRUE TRUEreplicate(n, expr) allows you to repeat a function many times, and return the answer in a vector
# returns 1 randomly sampled number from 1 to 10
sample(1:10,1)## [1] 8# let's repeat the above 10 times using replicate
replicate(10,sample(1:10,1))## [1] 10 5 10 5 2 9 2 4 3 5The next example shows how to write a function to do something, and then use the function inside replicate to repeat the function many times.
For example, we write a function to run a one-sample t-test on a random sample drawn from a normal distribution
ttest_result <- function(){
sample <- rnorm(10,0,1)
t_out <- t.test(sample, mu=0)
return(t_out$statistic)
}
# get 10 t-values from repeating the above 10 times
replicate(10, ttest_result() )## t t t t t t
## 0.4534718 0.9607405 -1.5162592 -1.4030149 -0.3533028 0.9144664
## t t t t
## 1.4682818 -0.7693148 -0.2584975 1.2003557The apply family of functions can be used to “apply” a function across elements of an object. A general overview can be found here
Some of the apply functions include: apply(), lapply, and sapply.
Here is part of the definition of lapply from the help file:
lapply returns a list of the same length as X, each element of which is the result of applying FUN to the corresponding element of X.
Let’s see some examples:
Let’s apply a function to each of the elements in a vector. To keep things simple, our function will add 1 to a number
some_numbers <- c(1,2,3,4)
add_one <- function(x){
return(x+1)
}
# returns a list, containing the answers
lapply(some_numbers, add_one)## [[1]]
## [1] 2
##
## [[2]]
## [1] 3
##
## [[3]]
## [1] 4
##
## [[4]]
## [1] 5# unlists the list
unlist(lapply(some_numbers,add_one))## [1] 2 3 4 5# sapply does the unlisting for you
sapply(some_numbers, add_one)## [1] 2 3 4 5An alternative syntax for lapply and sapply let’s you define the function you want to apply inside the lapply or sapply function.
In this case, each element in the vector some_numbers will become the x value in the function.
some_numbers <- c(1,2,3,4)
lapply(some_numbers, FUN = function(x){x+1})## [[1]]
## [1] 2
##
## [[2]]
## [1] 3
##
## [[3]]
## [1] 4
##
## [[4]]
## [1] 5sapply(some_numbers, FUN = function(x){x+1})## [1] 2 3 4 5The apply function can be used on 2-dimensional data, and allows you to apply a function across the rows or the columns of the data.
Let’s say you had a 5x5 matrix of random numbers. Let’s find the sum of each row
random_matrix <- matrix(sample(1:10,25, replace=TRUE),ncol=5)
# applies the sum function to each row
# 1 tells apply to go across rows
apply(random_matrix,1,sum)## [1] 27 19 29 25 27The sum of each column
# applies the sum function to each column
# 2 tells apply to go across columns
apply(random_matrix, 2, sum)## [1] 23 34 26 25 19Let’s say we have a matrix storing 3 samples. Each sample has 10 numbers. Each sample is stored in a column, and each row represents an observation.
sample_matrix <- matrix(rnorm(30,0,1),ncol=3)Let’s use apply to conduct 10 one-sample t-tests, one for each column. In this example, we can pass the mu=0 parameter into the t.test function. However, we will return the entire ouput of each t-test in a list.
apply(sample_matrix,2,t.test, mu=0)## [[1]]
##
## One Sample t-test
##
## data: newX[, i]
## t = 0.20558, df = 9, p-value = 0.8417
## alternative hypothesis: true mean is not equal to 0
## 95 percent confidence interval:
## -0.6487225 0.7784194
## sample estimates:
## mean of x
## 0.06484848
##
##
## [[2]]
##
## One Sample t-test
##
## data: newX[, i]
## t = -1.2098, df = 9, p-value = 0.2572
## alternative hypothesis: true mean is not equal to 0
## 95 percent confidence interval:
## -0.7980401 0.2418945
## sample estimates:
## mean of x
## -0.2780728
##
##
## [[3]]
##
## One Sample t-test
##
## data: newX[, i]
## t = -1.598, df = 9, p-value = 0.1445
## alternative hypothesis: true mean is not equal to 0
## 95 percent confidence interval:
## -0.9627990 0.1656493
## sample estimates:
## mean of x
## -0.3985749What if we wanted to return only the t-values, rather than whole output?
You might try this, but it doesn’t work
apply(sample_matrix,2,t.test$statistic, mu=0)## Error in t.test$statistic: object of type 'closure' is not subsettableSo, we write a custom function
apply(sample_matrix, 2,
FUN = function(x){
t_out <- t.test(x,mu=0)
return(t_out$statistic)
})## [1] 0.2055822 -1.2097766 -1.5980156