Vectorized code: calculating a log-sum
A common operation in statistics is to calculate the sum of log probabilities. The following code calculates the log-sum (the sum of the logs).
# x is a vector of probabilities
total <- 0
for(i in seq_along(x))
total <- total + log(x[i])
However this piece of code could be significantly improved using vectorized code.
Deze oefening maakt deel uit van de cursus
Writing Efficient R Code
Oefeninstructies
- Find the log-sum of
xusing thelog()andsum()functions, simplify the above loop. - Store your answer in the object
log_sum.
Praktische interactieve oefening
Probeer deze oefening eens door deze voorbeeldcode in te vullen.
# Initial code
n <- 100
total <- 0
x <- runif(n)
for(i in 1:n)
total <- total + log(x[i])
# Rewrite in a single line. Store the result in log_sum
log_sum <- ___