技能——如何优雅处理R循环中的error

先看例子：

value <- list(1, 2, 3, 'a', 5)
lapply(value, FUN = function(x){
  return(class(x))
})

可以看到这个列表里面元素的数据类型分别为：

[[1]]
[1] "numeric"

[[2]]
[1] "numeric"

[[3]]
[1] "numeric"

[[4]]
[1] "character"

[[5]]
[1] "numeric"

现在我们想求这个列表里面所有对象的对数值，代码：

log <- vector()
for (i in 1:5){
  log[i] = log(value[[i]])
}

最后运行出来的log向量为：

cat(log)
## 0 0.6931472 1.098612

可以看到log向量只有3个数值，原因在于当我们在执行这个循环的时候，有一次循环出现了错误（不能对字符串进行数学运算），导致了整个循环的提前终止，所以出现了这样的结果。但有的时候，我们希望循环不会受到某次循环错误的影响而提前终止，要继续运行下去，这就是如何优雅处理R循环中的error。

方法一：try()

我们首先来看一下，try()到底在干什么：

cal_log <- function(x){
  return(log(x))
}

for (i in 1:5){
  temp <- try(cal_log(value[[i]]), silent = T)
  print(class(temp))
}
##[1] "numeric"
##[1] "numeric"
##[1] "numeric"
##[1] "try-error"
##[1] "numeric"

实际上我们可以看到循环并没有因为log('a')而终止（因为第五个类型为numeric），那么我们是不是可以直接这样操作就得到我们的结果呢？

cal_log <- function(x){
  return(log(x))
}

log <- vector()

for (i in 1:5){
  log[i] <- try(cal_log(value[[i]]), silent = T)
}

print(log)
##[1] "0"                                                                
##[2] "0.693147180559945"                                                
##[3] "1.09861228866811"                                                 
##[4] "Error in log(x) : non-numeric argument to mathematical function\n"
##[5] "1.6094379124341" 

结果是不符合我们预期的，把原本的数值转换成了字符，所以我们可能需要把try()的输出个性化一下：注意到当运行出现error时，try()返回的对象类是特殊的"try-error"，所以我们可以这样来操作一下：

cal_log <- function(x){
  return(log(x))
}
log <- vector()
for (i in 1:5){
  temp <- try(cal_log(value[[i]]), silent = T)
  log[i] <- ifelse(class(temp) == "try-error", NA, temp)
}
print(log)
##[1] 0.0000000 0.6931472 1.0986123        NA 1.6094379

方法二：tryCatch()

tryCatch()函数对于我们大多数人来说，掌握下面的结构基本就够我们处理错误信息了：

tryCatch({
  <expr_1>
}, error = function(e){
  <expr_2>  #当出现error时就运行这句
}, warning = function(w){
  <expr_3>  #当出现warning时就运行这句
}, finally = {
  <expr_4>  #每次循环必运行，无论前面正常与否
})

那么我们可以把上面的操作用tryCatch()实现一下：

cal_log <- function(x){
  return(log(x))
}
log <- vector()
for (i in 1:5){
  tryCatch({
    log[i] = cal_log(value[[i]])
  }, error = function(e){
    log[i] = NA
    print('Encounter an error!')
  })
}
##[1] "Encounter an error!"
print(log)
##[1] 0.0000000 0.6931472 1.0986123        NA 1.6094379

这个时候，当循环运行到log("a")时，就会自动运行log[i] = NA，同时打印出提示信息：print('Encounter an error!')。

以上就是两种R中处理循环error的方法！

今天又是摸鱼的一天~~~~🐟🐟🐟