根据R中另一个单元格中的值用NA替换一个单元格

时间:2018-12-15 16:07:50

标签: r dataframe replace na

我有一个数据集,可以从中得到一个可重现的示例:

set.seed(1)
Data <- data.frame(
A = sample(0:5),
B = sample(0:5),
C = sample(0:5),
D = sample(0:5),
corr_A.B = sample(0:5),
corr_A.C = sample(0:5),
corr_A.D = sample(0:5))

> Data
  A B C D corr_A.B corr_A.C corr_A.D
1 1 5 4 2        1        2        4
2 5 3 1 3        5        5        0
3 2 2 3 4        0        1        2
4 3 0 5 0        4        0        1
5 0 4 2 1        2        3        3
6 4 1 0 5        3        4        5

我想检查B,C和D的每一列,如果它们的一个像元等于0,我想在同一行上用NA替换对应的corr_A列。例如,由于Data $ B [4]等于0,我希望将Data $ corr_A.B [4]替换为NA。

我希望获得以下结果:

> Data
  A B C D corr_A.B corr_A.C corr_A.D
1 1 5 4 2        1        2        4
2 5 3 1 3        5        5        0
3 2 2 3 4        0        1        2
4 3 0 5 0        NA       0       NA
5 0 4 2 1        2        3        3
6 4 1 0 5        3        NA       5

我尝试了不同的方法,使用for循环,但是我很努力。另外,在我正在处理的数据集中,有许多其他列不需要针对该条件进行检查,我希望能够专门指定要在哪个列中查找0值。

如果有人愿意尝试一下?非常感谢

5 个答案:

答案 0 :(得分:5)

使用函数is.na<-的单线。

is.na(Data[5:7]) <- Data[2:4] == 0

Data
#  A B C D corr_A.B corr_A.C corr_A.D
#1 1 5 4 2        1        2        4
#2 5 3 1 3        5        5        0
#3 2 2 3 4        0        1        2
#4 3 0 5 0       NA        0       NA
#5 0 4 2 1        2        3        3
#6 4 1 0 5        3       NA        5

答案 1 :(得分:2)

对于基本的R解决方案,我们可以在此处使用ifelse

Data$corr_A.B <- ifelse(Data$B == 0, NA, Data$corr_A.B)
Data$corr_A.C <- ifelse(Data$C == 0, NA, Data$corr_A.C)
Data$corr_A.D <- ifelse(Data$D == 0, NA, Data$corr_A.D)

答案 2 :(得分:1)

df<- data.frame(A=c(1,5,2,3,0,4),
                B=c(5,3,2,0,4,1),
                C=c(4,1,3,5,2,0),
                D=c(2,3,4,0,1,5),
                corr_A.B=c(1,5,0,4,2,3),
                corr_A.C=c(2,5,1,0,3,4),
                corr_A.D=c(4,0,2,1,3,5))

df %>% mutate(corr_A.B=case_when(B==0 ~ NA_real_,
                                 TRUE~ corr_A.B),
              corr_A.C=case_when(C==0 ~NA_real_,
                                  TRUE ~ corr_A.C),
              corr_A.D=case_when(D==0 ~ NA_real_,
                                  TRUE ~ corr_A.D))
A B C D corr_A.B corr_A.C corr_A.D
1 1 5 4 2        1        2        4
2 5 3 1 3        5        5        0
3 2 2 3 4        0        1        2
4 3 0 5 0       NA        0       NA
5 0 4 2 1        2        3        3
6 4 1 0 5        3       NA        5

答案 3 :(得分:1)

单线,矢量化但复杂的基本解决方案:

Data[t(t(which(Data[,2:4]==0,arr.ind=TRUE))+c(0,4))]<-NA

答案 4 :(得分:0)

使用apply()。您可以这样做:

cbind(Data,apply(Data[c("B","C","D")],2,function(x){
  ifelse(x==0,NA,x)
}))