ifel循環的有效替代方案

Question

會是一些如此善良並建議快速解決方案（矢量化和data.table）解決方案？

這是樣本數據和當前循環：

set.seed(123)
df <- data.frame(x=abs(rnorm(10)*100),y=abs(rnorm(10)*100))
df$ratio <- df$x/df$y

for(i in 1:nrow(df)) {
df[,c("x")][i] <- ifelse(df[,c("ratio")][i]>1.5,1.5*df[,c("y")][i],df[,c("x")][i])
}

因此，如果比率大於1.5我將替換列x值

Answer 1

嘗試

library(data.table)
setDT(df)[ratio > 1.5, x := 1.5*y]

或base R解決方案

transform(df, x=ifelse(ratio > 1.5, 1.5*y, x))

基准

set.seed(123)
df1 <- data.frame(x=abs(rnorm(1e6)*100),y=abs(rnorm(1e6)*100))
df1$ratio <-df1$x/df1$y
f1 <- function(){ as.data.table(df1)[ratio > 1.5, x:= 1.5*y]}

f2 <- function(){ indx <- df1$ratio > 1.5
                df1$x[indx] <- df1$y[indx]*1.5}

f3 <- function(){transform(df1, x=ifelse(ratio > 1.5, 1.5*y, x))}

#Another option suggested by @Steven Beaupré
f4 <- function(){mutate(df1, x=ifelse(ratio > 1.5, 1.5*y, x))}

 microbenchmark(f1(),f2(),f3(),f4(), unit='relative', times=40L)
 #Unit: relative
 #expr       min        lq     mean    median       uq      max neval cld
 #f1()  1.000000  1.000000 1.000000  1.000000 1.000000 1.000000    40 a  
 #f2()  2.829316  2.749836 2.047366  2.645489 1.301990 2.070973    40 b 
 #f3() 12.693416 12.954443 9.060991 12.689862 6.170528 7.935411    40 c
 #f4() 13.231567 13.300574 9.147105 12.636984 6.217343 6.286193    40 c

Answer 2

基地R的另一個選擇：

indx <- df$ratio > 1.5
df$x[indx] <- df$y[indx] * 1.5

即使數據集相對較大，這也可能非常快。