使用数据帧列中的条件在R中生成向量

Question

我有以下数组：

  Year   Month   Day   Hour   
1   1      1       1    0
2   1      1       1    3
...

等等

我编写了一个函数，然后尝试使用apply进行矢量化，以便逐行运行计算，但是由于布尔值，该函数不起作用：

day_in_season<-function(tarr){
  #first month in season
  if((tarr$month==12) || (tarr$month==3) ||(tarr$month==6) || (tarr$month==9)){
    d=tarr$day
  #second month in season
  }else if ((tarr$month==1) || (tarr$month==4)){
    d=31+tarr$day
  }else if((tarr$month==7) || (tarr$month==10)){
    d=30+tarr$day
  #third month in season
  }else if((tarr$month==2)){
    d=62+tarr$day
  }else{
    d=61+tarr$day
  }
  h=tarr$hour/24
  d=d+h
  return(d)
}

我试过了

apply(tdjf,1,day_in_season)

但它引发了以下异常：

tarr $ month中的错误：$运算符对原子向量无效

（我已经知道这个潜在的陷阱，但这就是为什么我想首先使用apply的原因！）

目前，使它正常工作的唯一方法是：

days<-c()
for (x in 1:nrow(tdjf)){
  d<-day_in_season(tdjf[x,])
  days=append(days,d)
}

如果只有几个值，我会举起手来，仅使用for循环，这会降低效率，但是我有15,000多行，那只是一个数据集。 我知道必须有一种方法使其起作用。

Answer 1

要向量化代码，请使用ifelse()和| 代替|| ：

ifelse(
        (tarr$month==12) | (tarr$month==3) |(tarr$month==6) | (tarr$month==9),
        tarr$day, 
        ifelse((tarr$month==1) | (tarr$month==4), 
               31+tarr$day, 
               ifelse((tarr$month==7) | (tarr$month==10),
                      30+tarr$day,
                      ifelse(tarr$month==2, 
                             62+tarr$day,
                             61+tarr$day)
                       )
                )
        )+tarr$hour/24

Answer 2

您可能会对构造良好的for循环可以运行的速度感到惊讶。 如果设计得当，它将具有与apply语句几乎相同的效率。

在你的情况下适当的for循环是

tdjf$days <- vector ("numeric", nrow (tdjf))
for (x in seq_along (tdjf$days)){
  tdjf$days [x] <- day_in_season(tdjf[x,])
}

如果您真的想走应用路线，我建议您重写函数以接受三个参数（ month ， day和hour ）并将这三mapply