R 中带有 ifelse 语句的 for 循环

Question

我有一个数据集，其中包含物种属 GEN 的 as.factor 类型数据和衰减水平 DECAY 我想使用来自这 2 列的数据计算每一行的 AGB 仅使用 GEN 的第一组代码成功运行创建 AGB 列存储每个样本的 AGB 值。 第二组代码没有运行。 我省略了另外两个属以避免在这里重复。 错误粘贴在最后。 关于为什么相同的方法在添加条件时不起作用的任何提示。

for (i in 1:nrow(dat)){
  if(dat$GEN[i] =='Avicennia'){
    dat$AGB[i]<- 140*dat$dbh[i]^2.44
  }else if (dat$GEN[i]=='Laguncularia'){
    dat$AGB[i]<- 100.23*dat$dbh[i]^2.53
  }else if (dat$GEN[i]=='Rhizophora'){
    dat$AGB[i]<-128.23*dat$dbh[i]^2.61
  }
}

for (i in 1:nrow(dat)){
  if(dat$GEN[i] =='Avicennia'& dat$DECAY[i]==0){
    dat$AGB[i]<- 0.14*dat$dbh[i]^2.4
  }else if (dat$GEN[i]=='Avicennia'& dat$DECAY[i]==0.5){
    dat$AGB[i]<- 0.9875*0.14*dat$dbh[i]^2.4
  }else if (dat$GEN[i]=='Avicennia' & dat$DECAY[i]==1){
    dat$AGB[i]<-0.975*0.14*dat$dbh[i]^2.4
  }else if (dat$GEN[i]=='Avicennia'& dat$DECAY[i]==2){
    dat$AGB[i]<-0.8*0.14*dat$dbh[i]^2.4
  }else if (dat$GEN[i]=='Avicennia'& dat$DECAY[i]==3){
    dat$AGB[i]<-0.070*dat$dbh[i]^2.4
  }
}

Error in if (dat$GEN[i] == "Avicennia" & dat$DECAY[i] == 0) { : 
  missing value where TRUE/FALSE needed

Answer 1

如果我们想使用紧凑版本， ifelse （嵌套）或case_when会更合适

library(dplyr)
dat %>%
   mutate(AGB = case_when(
              GEN == 'Avicennia' & DECAY == 0 & !is.na(DECAY) ~ 0.14* dbh^2.4,
              GEN == 'Avicennia' & DECAY == 0.5 & !is.na(DECAY) ~ 0.9875 *0.14* dbh^2.4,
              GEN == 'Avicennia' & DECAY == 1 & !is.na(DECAY) ~ 0.975 *0.14* dbh^2.4,
              GEN == 'Avicennia' & DECAY == 2 & !is.na(DECAY) ~ 0.8 *0.14* dbh^2.4,
              GEN == 'Avicennia' & DECAY == 3 & !is.na(DECAY) ~ 0.07 *0.14* dbh^2.4,
              TRUE ~ as.numeric(AGB))

注意：当我们尝试将 ( == ) 与浮动值（即 0.5）进行比较时，总是存在风险。 确保值准确且不涉及任何精度

---

另一种选择是使用命名向量来匹配 'DECAY' 中的值，然后根据匹配值进行计算

nm1 <- setNames(c(1, 0.9875, 0.975, 0.8, 0.07), c(0, 0.5, 1, 2, 3))
i1 <- dat$GEN == 'Avicennia'
dat$AGB[i1] <- with(dat, nm1[as.character(DECAY[i1])] * 0.14 * dbh[i1]^2.4)