заявление ifelse в mapply r

Question

Я использую mapply, чтобы найти сумму диапазона индексов для всех случаев:

score = mapply(function(x, y, z) sum(df[x, y:z]), seq_len(nrow(df)), df$index, df$index+10)

Как добавить оператор ifelse в mapply, чтобы он применял функцию, только если df $ index> 5, иначе сумма равна 0?

Answer 1

Это, вероятно, должно работать

mapply(function(x, y, z) if (y > 5) sum(df[x, y:z]) else 0, 
         seq_len(nrow(df)), df$index, df$index+10)

Или вы можете избежать if / else, умножив на (y > 5), поэтому index > 5 будет умножено на 1 (TRUE), давая sum(df[x, y:z]), а index <= 5 будет умножено на 0 (* 1011) *) давая 0.

mapply(function(x, y, z) sum(df[x, y:z]) * (y > 5), 
        seq_len(nrow(df)), df$index, df$index+10)

Answer 2

Просто оберните ifelse, и оно должно работать

mapply(function(x, y, z) ifelse(y > 5, sum(df[x, y:z]), 0), 
             seq_len(nrow(df)), df$index, df$index+10)

---

Лучшим вариантом будет предварительное назначение на основе значения 'index' для 0 для тех 'index', которые меньше чемили равно 5. Должно быть более эффективным

i1 <- df$index > 5
score[!i1] <- 0
score[i1] <- mapply(function(x, y, z) sum(df[x, y:z]),
                     seq_len(nrow(df))[i1], df$index[i1], df$index[i1] + 10)