KC2*_*200 5 if-statement r lapply dataframe
我想对数据帧中的每个列(ad)应用if else语句,如果每个值都满足基于另一列(m)的条件,则该值保持不变,否则该值更改为零。
这是一个可重现的示例:
a <- c(10, 11, 12, 15,7,8)
b <- c(15,11,16,17,1.5,9)
c <- c(18,14,6,14,1,17)
d <- c(12,11,13,19,2,4)
p <- c(2,1,3,2.5,1.3,2.1)
m <- c(12,13,8,9,14,11)
df<- data.frame(a,b,c,d,p,m)
print(df)
a b c d p m
1 10 15.0 18 12 2.0 12
2 11 11.0 14 11 1.0 13
3 12 16.0 6 13 3.0 8
4 15 17.0 14 19 2.5 9
5 7 1.5 1 2 1.3 14
6 8 9.0 17 4 2.1 11
Run Code Online (Sandbox Code Playgroud)
这是我要执行的操作的“长篇”版本,该版本有效,但非常重复:
df$a<- ifelse((df$a-df$p)>df$m, df$a, 0)
df$b<- ifelse((df$b-df$p)>df$m, df$b, 0)
df$c<- ifelse((df$c-df$p)>df$m, df$c, 0)
df$d<- ifelse((df$d-df$p)>df$m, df$d, 0)
print(df)
Run Code Online (Sandbox Code Playgroud)
这是我想要的输出:
a b c d p m
1 0 15 18 0 2.0 12
2 11 11 14 11 1.0 7
3 12 16 0 13 3.0 8
4 15 17 14 19 2.5 9
5 0 0 0 0 1.3 14
6 0 0 17 0 2.1 11
Run Code Online (Sandbox Code Playgroud)
我们可以使用 lapply
df[1:4] <- lapply(df[1:4], function(x) ifelse((x - df$p) > df$m, x, 0))
df
# a b c d p m
#1 0 15 18 0 2.0 12
#2 0 0 0 0 1.0 13
#3 12 16 0 13 3.0 8
#4 15 17 14 19 2.5 9
#5 0 0 0 0 1.3 14
#6 0 0 17 0 2.1 11
Run Code Online (Sandbox Code Playgroud)
还是不使用 ifelse
df[1:4] <- ((df[1:4]-df$p) > df$m)*df[1:4]
Run Code Online (Sandbox Code Playgroud)