CommencerCommencer gratuitement

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.

Cet exercice fait partie du cours

Writing Efficient R Code

Afficher le cours

Instructions

  • Find the log-sum of x using the log() and sum() functions, simplify the above loop.
  • Store your answer in the object log_sum.

Exercice interactif pratique

Essayez cet exercice en complétant cet exemple de code.

# 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 <- ___
Modifier et exécuter le code