Introduction to vectorization and loop functionals.

**Before class, you can prepare by reading the following materials:**

Material for this lecture was borrowed and adopted from

- https://rdpeng.github.io/Biostat776/lecture-loop-functions.html
- https://rafalab.github.io/dsbook/programming-basics.html#vectorization

**At the end of this lesson you will:**

- Understand how to perform vector arithmetics in R
- Implement the 5 functional loops in R (vs e.g. for loops) in R

Writing `for`

and `while`

loops are useful and easy to understand, but in R we rarely use them. As you learn more R, you will realize that **vectorization** is preferred over for-loops since it results in shorter and clearer code.

In R, arithmetic operations on **vectors occur element-wise**. For a quick example, suppose we have height in inches:

```
inches <- c(69, 62, 66, 70, 70, 73, 67, 73, 67, 70)
```

and want to convert to centimeters. Notice what happens when we multiply inches by 2.54:

```
inches * 2.54
```

```
[1] 175.26 157.48 167.64 177.80 177.80 185.42 170.18 185.42 170.18
[10] 177.80
```

In the line above, we multiplied each element by 2.54. Similarly, if for each entry we want to compute how many inches taller or shorter than 69 inches (the average height for males), we can subtract it from every entry like this:

```
inches - 69
```

` [1] 0 -7 -3 1 1 4 -2 4 -2 1`

If we have two vectors of the same length, and we sum them in R, they will be added entry by entry as follows:

```
x <- 1:10
y <- 1:10
x + y
```

` [1] 2 4 6 8 10 12 14 16 18 20`

The same holds for other mathematical operations, such as `-`

, `*`

and `/`

.

```
x <- 1:10
sqrt(x)
```

```
[1] 1.000000 1.414214 1.732051 2.000000 2.236068 2.449490 2.645751
[8] 2.828427 3.000000 3.162278
```

```
y <- 1:10
x*y
```

` [1] 1 4 9 16 25 36 49 64 81 100`

While `for`

loops are perfectly valid, when you use vectorization in an element-wise fashion, there is no need for `for`

loops because we can apply what are called functional loops.

**Functional loops** are functions that help us apply the same function to each entry in a vector, matrix, data frame, or list. Here are a list of them:

`lapply()`

: Loop over a list and evaluate a function on each element`sapply()`

: Same as`lapply`

but try to simplify the result`apply()`

: Apply a function over the margins of an array`tapply()`

: Apply a function over subsets of a vector`mapply()`

: Multivariate version of`lapply`

An auxiliary function `split()`

is also useful, particularly in conjunction with `lapply()`

.

`lapply()`

The `lapply()`

function does the following simple series of operations:

- it loops over a list, iterating over each element in that list
- it applies a
*function*to each element of the list (a function that you specify) - and returns a list (the
`l`

in`lapply()`

is for “list”).

This function takes three arguments: (1) a list `X`

; (2) a function (or the name of a function) `FUN`

; (3) other arguments via its `...`

argument. If `X`

is not a list, it will be coerced to a list using `as.list()`

.

The body of the `lapply()`

function can be seen here.

```
lapply
```

```
function (X, FUN, ...)
{
FUN <- match.fun(FUN)
if (!is.vector(X) || is.object(X))
X <- as.list(X)
.Internal(lapply(X, FUN))
}
<bytecode: 0x7f9a1c0461a0>
<environment: namespace:base>
```

**Note**: the actual looping is done internally in C code for efficiency reasons.

It is important to remember that `lapply()`

always returns a list, regardless of the class of the input.

Here’s an example of applying the `mean()`

function to all elements of a list. If the original list has names, the the names will be preserved in the output.

Notice that here we are passing the `mean()`

function as an argument to the `lapply()`

function. Functions in R can be used this way and can be passed back and forth as arguments just like any other object. When you pass a function to another function, you do not need to include the open and closed parentheses `()`

like you do when you are *calling* a function.

Here is another example of using `lapply()`

.

```
$a
[1] 2.5
$b
[1] 0.248845
$c
[1] 0.9935285
$d
[1] 5.051388
```

You can use `lapply()`

to evaluate a function multiple times each with a different argument. Below, is an example where I call the `runif()`

function (to generate uniformly distributed random variables) four times, each time generating a different number of random numbers.

```
x <- 1:4
lapply(x, runif)
```

```
[[1]]
[1] 0.02778712
[[2]]
[1] 0.5273108 0.8803191
[[3]]
[1] 0.37306337 0.04795913 0.13862825
[[4]]
[1] 0.3214921 0.1548316 0.1322282 0.2213059
```

When you pass a function to `lapply()`

, `lapply()`

takes elements of the list and passes them as the *first argument* of the function you are applying. In the above example, the first argument of `runif()`

is `n`

, and so the elements of the sequence `1:4`

all got passed to the `n`

argument of `runif()`

.

Functions that you pass to `lapply()`

may have other arguments. For example, the `runif()`

function has a `min`

and `max`

argument too. In the example above I used the default values for `min`

and `max`

. How would you be able to specify different values for that in the context of `lapply()`

?

Here is where the `...`

argument to `lapply()`

comes into play. Any arguments that you place in the `...`

argument will get passed down to the function being applied to the elements of the list.

Here, the `min = 0`

and `max = 10`

arguments are passed down to `runif()`

every time it gets called.

```
x <- 1:4
lapply(x, runif, min = 0, max = 10)
```

```
[[1]]
[1] 2.263808
[[2]]
[1] 1.314165 9.815635
[[3]]
[1] 3.270137 5.069395 6.814425
[[4]]
[1] 0.9916910 1.1890256 0.5043966 9.2925392
```

So now, instead of the random numbers being between 0 and 1 (the default), the are all between 0 and 10.

The `lapply()`

function (and its friends) makes heavy use of *anonymous* functions. Anonymous functions are like members of Project Mayhem—they have no names. These functions are generated “on the fly” as you are using `lapply()`

. Once the call to `lapply()`

is finished, the function disappears and does not appear in the workspace.

Here I am creating a list that contains two matrices.

```
$a
[,1] [,2]
[1,] 1 3
[2,] 2 4
$b
[,1] [,2]
[1,] 1 4
[2,] 2 5
[3,] 3 6
```

Suppose I wanted to extract the first column of each matrix in the list. I could write an anonymous function for extracting the first column of each matrix.

```
lapply(x, function(elt) { elt[,1] })
```

```
$a
[1] 1 2
$b
[1] 1 2 3
```

Notice that I put the `function()`

definition right in the call to `lapply()`

. This is perfectly legal and acceptable. You can put an arbitrarily complicated function definition inside `lapply()`

, but if it’s going to be more complicated, it’s probably a better idea to define the function separately.

For example, I could have done the following.

```
f <- function(elt) {
elt[, 1]
}
lapply(x, f)
```

```
$a
[1] 1 2
$b
[1] 1 2 3
```

Now the function is no longer anonymous; its name is `f`

. Whether you use an anonymous function or you define a function first depends on your context. If you think the function `f`

is something you are going to need a lot in other parts of your code, you might want to define it separately. But if you are just going to use it for this call to `lapply()`

, then it is probably simpler to use an anonymous function.

`sapply()`

The `sapply()`

function behaves similarly to `lapply()`

; the only real difference is in the return value. `sapply()`

will try to simplify the result of `lapply()`

if possible. Essentially, `sapply()`

calls `lapply()`

on its input and then applies the following algorithm:

If the result is a list where every element is length 1, then a vector is returned

If the result is a list where every element is a vector of the same length (> 1), a matrix is returned.

If it can’t figure things out, a list is returned

Here’s the result of calling `lapply()`

.

```
$a
[1] 2.5
$b
[1] -0.251483
$c
[1] 1.481246
$d
[1] 4.968715
```

Notice that `lapply()`

returns a list (as usual), but that each element of the list has length 1.

Here’s the result of calling `sapply()`

on the same list.

```
sapply(x, mean)
```

```
a b c d
2.500000 -0.251483 1.481246 4.968715
```

Because the result of `lapply()`

was a list where each element had length 1, `sapply()`

collapsed the output into a numeric vector, which is often more useful than a list.

`split()`

The `split()`

function takes a vector or other objects and splits it into groups determined by a factor or list of factors.

The arguments to `split()`

are

```
str(split)
```

`function (x, f, drop = FALSE, ...) `

where

`x`

is a vector (or list) or data frame`f`

is a factor (or coerced to one) or a list of factors`drop`

indicates whether empty factors levels should be dropped

The combination of `split()`

and a function like `lapply()`

or `sapply()`

is a common paradigm in R. The basic idea is that you can take a data structure, split it into subsets defined by another variable, and apply a function over those subsets. The results of applying that function over the subsets are then collated and returned as an object. This sequence of operations is sometimes referred to as “map-reduce” in other contexts.

Here we simulate some data and split it according to a factor variable. Note that we use the `gl()`

function to “generate levels” in a factor variable.

```
$`1`
[1] 0.3981302 -0.4075286 1.3242586 -0.7012317 -0.5806143 -1.0010722
[7] -0.6681786 0.9451850 0.4337021 1.0051592
$`2`
[1] 0.34822440 0.94893818 0.64667919 0.03527777 0.59644846 0.41531800
[7] 0.07689704 0.52804888 0.96233331 0.70874005
$`3`
[1] 1.13444766 1.76559900 1.95513668 0.94943430 0.69418458
[6] 1.89367370 -0.04729815 2.97133739 0.61636789 2.65414530
```

A common idiom is `split`

followed by an `lapply`

.

```
Ozone Solar.R Wind Temp Month Day
1 41 190 7.4 67 5 1
2 36 118 8.0 72 5 2
3 12 149 12.6 74 5 3
4 18 313 11.5 62 5 4
5 NA NA 14.3 56 5 5
6 28 NA 14.9 66 5 6
```

We can split the `airquality`

data frame by the `Month`

variable so that we have separate sub-data frames for each month.

```
List of 5
$ 5:'data.frame': 31 obs. of 6 variables:
..$ Ozone : int [1:31] 41 36 12 18 NA 28 23 19 8 NA ...
..$ Solar.R: int [1:31] 190 118 149 313 NA NA 299 99 19 194 ...
..$ Wind : num [1:31] 7.4 8 12.6 11.5 14.3 14.9 8.6 13.8 20.1 8.6 ...
..$ Temp : int [1:31] 67 72 74 62 56 66 65 59 61 69 ...
..$ Month : int [1:31] 5 5 5 5 5 5 5 5 5 5 ...
..$ Day : int [1:31] 1 2 3 4 5 6 7 8 9 10 ...
$ 6:'data.frame': 30 obs. of 6 variables:
..$ Ozone : int [1:30] NA NA NA NA NA NA 29 NA 71 39 ...
..$ Solar.R: int [1:30] 286 287 242 186 220 264 127 273 291 323 ...
..$ Wind : num [1:30] 8.6 9.7 16.1 9.2 8.6 14.3 9.7 6.9 13.8 11.5 ...
..$ Temp : int [1:30] 78 74 67 84 85 79 82 87 90 87 ...
..$ Month : int [1:30] 6 6 6 6 6 6 6 6 6 6 ...
..$ Day : int [1:30] 1 2 3 4 5 6 7 8 9 10 ...
$ 7:'data.frame': 31 obs. of 6 variables:
..$ Ozone : int [1:31] 135 49 32 NA 64 40 77 97 97 85 ...
..$ Solar.R: int [1:31] 269 248 236 101 175 314 276 267 272 175 ...
..$ Wind : num [1:31] 4.1 9.2 9.2 10.9 4.6 10.9 5.1 6.3 5.7 7.4 ...
..$ Temp : int [1:31] 84 85 81 84 83 83 88 92 92 89 ...
..$ Month : int [1:31] 7 7 7 7 7 7 7 7 7 7 ...
..$ Day : int [1:31] 1 2 3 4 5 6 7 8 9 10 ...
$ 8:'data.frame': 31 obs. of 6 variables:
..$ Ozone : int [1:31] 39 9 16 78 35 66 122 89 110 NA ...
..$ Solar.R: int [1:31] 83 24 77 NA NA NA 255 229 207 222 ...
..$ Wind : num [1:31] 6.9 13.8 7.4 6.9 7.4 4.6 4 10.3 8 8.6 ...
..$ Temp : int [1:31] 81 81 82 86 85 87 89 90 90 92 ...
..$ Month : int [1:31] 8 8 8 8 8 8 8 8 8 8 ...
..$ Day : int [1:31] 1 2 3 4 5 6 7 8 9 10 ...
$ 9:'data.frame': 30 obs. of 6 variables:
..$ Ozone : int [1:30] 96 78 73 91 47 32 20 23 21 24 ...
..$ Solar.R: int [1:30] 167 197 183 189 95 92 252 220 230 259 ...
..$ Wind : num [1:30] 6.9 5.1 2.8 4.6 7.4 15.5 10.9 10.3 10.9 9.7 ...
..$ Temp : int [1:30] 91 92 93 93 87 84 80 78 75 73 ...
..$ Month : int [1:30] 9 9 9 9 9 9 9 9 9 9 ...
..$ Day : int [1:30] 1 2 3 4 5 6 7 8 9 10 ...
```

Then we can take the column means for `Ozone`

, `Solar.R`

, and `Wind`

for each sub-data frame.

```
$`5`
Ozone Solar.R Wind
NA NA 11.62258
$`6`
Ozone Solar.R Wind
NA 190.16667 10.26667
$`7`
Ozone Solar.R Wind
NA 216.483871 8.941935
$`8`
Ozone Solar.R Wind
NA NA 8.793548
$`9`
Ozone Solar.R Wind
NA 167.4333 10.1800
```

Using `sapply()`

might be better here for a more readable output.

```
5 6 7 8 9
Ozone NA NA NA NA NA
Solar.R NA 190.16667 216.483871 NA 167.4333
Wind 11.62258 10.26667 8.941935 8.793548 10.1800
```

Unfortunately, there are `NA`

s in the data so we cannot simply take the means of those variables. However, we can tell the `colMeans`

function to remove the `NA`

s before computing the mean.

```
5 6 7 8 9
Ozone 23.61538 29.44444 59.115385 59.961538 31.44828
Solar.R 181.29630 190.16667 216.483871 171.857143 167.43333
Wind 11.62258 10.26667 8.941935 8.793548 10.18000
```

Occasionally, we may want to split an R object according to levels defined in more than one variable. We can do this by creating an interaction of the variables with the `interaction()`

function.

```
[1] 1 1 1 1 1 2 2 2 2 2
Levels: 1 2
```

```
f2
```

```
[1] 1 1 2 2 3 3 4 4 5 5
Levels: 1 2 3 4 5
```

```
## Create interaction of two factors
interaction(f1, f2)
```

```
[1] 1.1 1.1 1.2 1.2 1.3 2.3 2.4 2.4 2.5 2.5
Levels: 1.1 2.1 1.2 2.2 1.3 2.3 1.4 2.4 1.5 2.5
```

With multiple factors and many levels, creating an interaction can result in many levels that are empty.

```
List of 10
$ 1.1: num [1:2] 1.512 0.083
$ 2.1: num(0)
$ 1.2: num [1:2] 0.567 -1.025
$ 2.2: num(0)
$ 1.3: num 0.323
$ 2.3: num 1.04
$ 1.4: num(0)
$ 2.4: num [1:2] 0.0991 -0.4541
$ 1.5: num(0)
$ 2.5: num [1:2] -0.6558 -0.0359
```

Notice that there are 4 categories with no data. But we can drop empty levels when we call the `split()`

function.

```
List of 6
$ 1.1: num [1:2] 1.512 0.083
$ 1.2: num [1:2] 0.567 -1.025
$ 1.3: num 0.323
$ 2.3: num 1.04
$ 2.4: num [1:2] 0.0991 -0.4541
$ 2.5: num [1:2] -0.6558 -0.0359
```

`tapply()`

is used to apply a function over subsets of a vector. It can be thought of as a combination of `split()`

and `sapply()`

for vectors only. I’ve been told that the “t” in `tapply()`

refers to “table”, but that is unconfirmed.

```
str(tapply)
```

`function (X, INDEX, FUN = NULL, ..., default = NA, simplify = TRUE) `

The arguments to `tapply()`

are as follows:

`X`

is a vector`INDEX`

is a factor or a list of factors (or else they are coerced to factors)`FUN`

is a function to be applied- … contains other arguments to be passed
`FUN`

`simplify`

, should we simplify the result?

Given a vector of numbers, one simple operation is to take group means.

```
## Simulate some data
x <- c(rnorm(10), runif(10), rnorm(10, 1))
## Define some groups with a factor variable
f <- gl(3, 10)
f
```

```
[1] 1 1 1 1 1 1 1 1 1 1 2 2 2 2 2 2 2 2 2 2 3 3 3 3 3 3 3 3 3 3
Levels: 1 2 3
```

```
tapply(x, f, mean)
```

```
1 2 3
0.1896235 0.5336667 0.9568236
```

We can also take the group means without simplifying the result, which will give us a list. For functions that return a single value, usually, this is not what we want, but it can be done.

```
tapply(x, f, mean, simplify = FALSE)
```

```
$`1`
[1] 0.1896235
$`2`
[1] 0.5336667
$`3`
[1] 0.9568236
```

We can also apply functions that return more than a single value. In this case, `tapply()`

will not simplify the result and will return a list. Here’s an example of finding the range of each sub-group.

```
tapply(x, f, range)
```

```
$`1`
[1] -1.869789 1.497041
$`2`
[1] 0.09515213 0.86723879
$`3`
[1] -0.5690822 2.3644349
```

`apply()`

The `apply()`

function is used to a evaluate a function (often an anonymous one) over the margins of an array. It is most often used to apply a function to the rows or columns of a matrix (which is just a 2-dimensional array). However, it can be used with general arrays, for example, to take the average of an array of matrices. Using `apply()`

is not really faster than writing a loop, but it works in one line and is highly compact.

```
str(apply)
```

`function (X, MARGIN, FUN, ..., simplify = TRUE) `

The arguments to `apply()`

are

`X`

is an array`MARGIN`

is an integer vector indicating which margins should be “retained”.`FUN`

is a function to be applied`...`

is for other arguments to be passed to`FUN`

Here I create a 20 by 10 matrix of Normal random numbers. I then compute the mean of each column.

```
[1] 0.02218266 -0.15932850 0.09021391 0.14723035 -0.22431309
[6] -0.49657847 0.30095015 0.07703985 -0.20818099 0.06809774
```

I can also compute the sum of each row.

```
apply(x, 1, sum) ## Take the mean of each row
```

```
[1] -0.48483448 5.33222301 -3.33862932 -1.39998450 2.37859098
[6] 0.01082604 -6.29457190 -0.26287700 0.71133578 -3.38125293
[11] -4.67522818 3.01900232 -2.39466347 -2.16004389 5.33063755
[16] -2.92024635 3.52026401 -1.84880901 -4.10213912 5.30667310
```

Note that in both calls to `apply()`

, the return value was a vector of numbers.

You’ve probably noticed that the second argument is either a 1 or a 2, depending on whether we want row statistics or column statistics. What exactly *is* the second argument to `apply()`

?

The `MARGIN`

argument essentially indicates to `apply()`

which dimension of the array you want to preserve or retain. So when taking the mean of each column, I specify

```
apply(x, 2, mean)
```

because I want to collapse the first dimension (the rows) by taking the mean and I want to preserve the number of columns. Similarly, when I want the row sums, I run

```
apply(x, 1, mean)
```

because I want to collapse the columns (the second dimension) and preserve the number of rows (the first dimension).

For the special case of column/row sums and column/row means of matrices, we have some useful shortcuts.

`rowSums`

=`apply(x, 1, sum)`

`rowMeans`

=`apply(x, 1, mean)`

`colSums`

=`apply(x, 2, sum)`

`colMeans`

=`apply(x, 2, mean)`

The shortcut functions are heavily optimized and hence are *much* faster, but you probably won’t notice unless you’re using a large matrix. Another nice aspect of these functions is that they are a bit more descriptive. It’s arguably more clear to write `colMeans(x)`

in your code than `apply(x, 2, mean)`

.

You can do more than take sums and means with the `apply()`

function. For example, you can compute quantiles of the rows of a matrix using the `quantile()`

function.

```
[,1] [,2] [,3] [,4] [,5] [,6]
25% -1.0884151 -0.6693040 0.2908481 -0.4602083 -1.0432010 -1.12773555
75% 0.1843547 0.8210295 1.3667301 0.4424153 0.3571219 0.03653687
[,7] [,8] [,9] [,10] [,11] [,12]
25% -1.4571706 -0.2406991 -0.3226845 -0.329898 -0.8677524 -0.2023664
75% -0.1705336 0.6504486 1.1460854 1.247092 0.4138139 0.9145331
[,13] [,14] [,15] [,16] [,17]
25% -0.9796050 -1.3551031 -0.1823252 -1.260911898 -0.9954289
75% 0.5448777 -0.5396766 0.7795571 0.002908451 0.4323192
[,18] [,19] [,20]
25% -0.3767354 -0.8557544 -0.7000363
75% 0.7542638 0.5440158 0.5432995
```

Notice that I had to pass the `probs = c(0.25, 0.75)`

argument to `quantile()`

via the `...`

argument to `apply()`

.

For a higher dimensional example, I can create an array of \(2\times2\) matrices and then compute the average of the matrices in the array.

```
[,1] [,2]
[1,] 0.1681387 -0.1039673
[2,] 0.3519741 -0.4029737
```

In the call to `apply()`

here, I indicated via the `MARGIN`

argument that I wanted to preserve the first and second dimensions and to collapse the third dimension by taking the mean.

There is a faster way to do this specific operation via the `colMeans()`

function.

```
rowMeans(a, dims = 2) ## Faster
```

```
[,1] [,2]
[1,] 0.1681387 -0.1039673
[2,] 0.3519741 -0.4029737
```

In this situation, I might argue that the use of `rowMeans()`

is less readable, but it is substantially faster with large arrays.

`mapply()`

The `mapply()`

function is a multivariate apply of sorts which applies a function in parallel over a set of arguments. Recall that `lapply()`

and friends only iterate over a single R object. What if you want to iterate over multiple R objects in parallel? This is what `mapply()`

is for.

```
str(mapply)
```

`function (FUN, ..., MoreArgs = NULL, SIMPLIFY = TRUE, USE.NAMES = TRUE) `

The arguments to `mapply()`

are

`FUN`

is a function to apply`...`

contains R objects to apply over`MoreArgs`

is a list of other arguments to`FUN`

.`SIMPLIFY`

indicates whether the result should be simplified

The `mapply()`

function has a different argument order from `lapply()`

because the function to apply comes first rather than the object to iterate over. The R objects over which we apply the function are given in the `...`

argument because we can apply over an arbitrary number of R objects.

For example, the following is tedious to type

`list(rep(1, 4), rep(2, 3), rep(3, 2), rep(4, 1))`

With `mapply()`

, instead we can do

```
mapply(rep, 1:4, 4:1)
```

```
[[1]]
[1] 1 1 1 1
[[2]]
[1] 2 2 2
[[3]]
[1] 3 3
[[4]]
[1] 4
```

This passes the sequence `1:4`

to the first argument of `rep()`

and the sequence `4:1`

to the second argument.

Here’s another example for simulating random Normal variables.

```
noise <- function(n, mean, sd) {
rnorm(n, mean, sd)
}
## Simulate 5 random numbers
noise(5, 1, 2)
```

`[1] -0.5196913 3.2979182 -0.6849525 1.7828267 2.7827545`

```
## This only simulates 1 set of numbers, not 5
noise(1:5, 1:5, 2)
```

`[1] -1.670517 2.796247 2.776826 5.351488 3.422804`

Here we can use `mapply()`

to pass the sequence `1:5`

separately to the `noise()`

function so that we can get 5 sets of random numbers, each with a different length and mean.

```
mapply(noise, 1:5, 1:5, 2)
```

```
[[1]]
[1] 0.8260273
[[2]]
[1] 4.764568 2.336980
[[3]]
[1] 4.6463819 2.5582108 0.9412167
[[4]]
[1] 3.978149 1.550018 -1.192223 6.338245
[[5]]
[1] 2.826182 1.347834 6.990564 4.976276 3.800743
```

The above call to `mapply()`

is the same as

```
list(noise(1, 1, 2), noise(2, 2, 2),
noise(3, 3, 2), noise(4, 4, 2),
noise(5, 5, 2))
```

```
[[1]]
[1] 0.644104
[[2]]
[1] 1.148037 3.993318
[[3]]
[1] 4.4553214 -0.4532612 3.7067970
[[4]]
[1] 5.4536273 5.3365220 -0.8486346 3.5292851
[[5]]
[1] 8.959267 6.593589 1.581448 1.672663 5.982219
```

The `mapply()`

function can be use to automatically “vectorize” a function. What this means is that it can be used to take a function that typically only takes single arguments and create a new function that can take vector arguments. This is often needed when you want to plot functions.

Here’s an example of a function that computes the sum of squares given some data, a mean parameter and a standard deviation. The formula is \(\sum_{i=1}^n(x_i-\mu)^2/\sigma^2\).

```
sumsq <- function(mu, sigma, x) {
sum(((x - mu) / sigma)^2)
}
```

This function takes a mean `mu`

, a standard deviation `sigma`

, and some data in a vector `x`

.

In many statistical applications, we want to minimize the sum of squares to find the optimal `mu`

and `sigma`

. Before we do that, we may want to evaluate or plot the function for many different values of `mu`

or `sigma`

. However, passing a vector of `mu`

s or `sigma`

s won’t work with this function because it’s not vectorized.

```
x <- rnorm(100) ## Generate some data
sumsq(1:10, 1:10, x) ## This is not what we want
```

`[1] 110.2594`

Note that the call to `sumsq()`

only produced one value instead of 10 values.

However, we can do what we want to do by using `mapply()`

.

```
[1] 196.2289 121.4765 108.3981 104.0788 102.1975 101.2393 100.6998
[8] 100.3745 100.1685 100.0332
```

There’s even a function in R called `Vectorize()`

that automatically can create a vectorized version of your function. So we could create a `vsumsq()`

function that is fully vectorized as follows.

```
[1] 196.2289 121.4765 108.3981 104.0788 102.1975 101.2393 100.6998
[8] 100.3745 100.1685 100.0332
```

Pretty cool, right?

The loop functions in R are very powerful because they allow you to conduct a series of operations on data using a compact form

The operation of a loop function involves iterating over an R object (e.g. a list or vector or matrix), applying a function to each element of the object, and the collating the results and returning the collated results.

Loop functions make heavy use of anonymous functions, which exist for the life of the loop function but are not stored anywhere

The

`split()`

function can be used to divide an R object in to subsets determined by another variable which can subsequently be looped over using loop functions.

Here are some post-lecture questions to help you think about the material discussed.

**Questions:**

- Write a function
`compute_s_n()`

that for any given`n`

computes the sum

\[ S_n = 1^2 + 2^2 + 3^2 + \ldots + n^2 \]

Report the value of the sum when \(n\) = 10.

Define an empty numerical vector

`s_n`

of size 25 using`s_n <- vector("numeric", 25)`

and store in the results of \(S_1, S_2, \ldots, S_n\) using a for-loop.Repeat Q3, but this time use

`sapply()`

.Plot

`s_n`

versus`n`

. Use points defined by \(n= 1, \ldots, 25\)

Text and figures are licensed under Creative Commons Attribution CC BY-NC-SA 4.0. The figures that have been reused from other sources don't fall under this license and can be recognized by a note in their caption: "Figure from ...".

For attribution, please cite this work as

Hicks (2021, Sept. 21). Statistical Computing: Vectorization and loop functionals. Retrieved from https://stephaniehicks.com/jhustatcomputing2021/posts/2021-09-21-loop-functions/

BibTeX citation

@misc{hicks2021vectorization, author = {Hicks, Stephanie}, title = {Statistical Computing: Vectorization and loop functionals}, url = {https://stephaniehicks.com/jhustatcomputing2021/posts/2021-09-21-loop-functions/}, year = {2021} }