基于行的值删除列

Question

给定两个数据框：

 C1<-c(3,4,4,4,5)
 C2<-c(3,7,3,4,5)
 C3<-c(5,6,3,7,4)
 DF<-data.frame(C1=C1,C2=C2,C3=C3)
 DF
   C1 C2 C3
1  3  3  5
2  4  7  6
3  4  3  3
4  4  4  7
5  5  5  4

并且

V1<-c(3,2,2,4,5)
V2<-c(3,7,3,5,2)
V3<-c(5,2,5,7,5)
V4<-c(1,1,2,3,4)
V5<-c(1,2,6,7,5)
DF2<-data.frame(V1=V1,V2=V2,V3=V3,V4=V4,V5=V5)
DF2
  V1 V2 V3 V4 V5
1  3  3  5  1  1
2  2  7  2  1  2
3  2  3  5  2  6
4  4  5  7  3  7
5  5  2  5  4  5

观察两个数据框中每一行的对应关系，C3列中的数值与我想要在DF2中同一行中删除的列数之间存在着一种关系。

C3列中的数值与需要在DF2中删除的列数之间的关系如下所示：

If C3≥7 drop V5
If C3=6.0:6.9 drop V4 and up (so basically V5,V4)
If C3=5.0:5.9 drop V3 and up (so basically V5,V4,V3)
If C3=4.0:4.9 drop V2 and up (so basically V5,V4,V3,V2)
If C3≤3.9 drop entire row

例如，根据C3的值，我希望DF2看起来像这样：

  V1 V2 V3 V4 V5
1  3  3
2  2  7  2

4  4  5  7  3 
5  5

我尝试编写一个简单的脚本来实现这个功能（因为我还是个新手，所以喜欢保持简单，以便清楚地看到发生了什么），但是我一直在遇到各种错误，所以我希望能得到有关如何继续的建议。

- Vinterwoo

---

您没有指定要删除哪一行（在哪个数据表中，假设它是 DF），当 C3<=3.9 时。 - Subs

Answer 1

也许最简单的方法是这样的：

DF3 <- DF2
for (i in seq_len(nrow(DF3))) {
  DF3[i, seq_len(ncol(DF3)) >= DF[i, ]$C3 - 2] <- NA
}
DF3

那么，接下来，

> DF3
  V1 V2 V3 V4 V5
1  3  3 NA NA NA
2  2  7  2 NA NA
3 NA NA NA NA NA
4  4  5  7  3 NA
5  5 NA NA NA NA

- kohske

Answer 2

我喜欢Koshke的回答，但是如果你设为NA的规则没有很好的数学性质，或者需要任意定义规则，那么这种方法可以给你更大的灵活性。首先，定义一个函数来根据你的规则返回要删除的列：

f <- function(x) {
  if(x >= 7){
    out <- 5
  }else if(x >= 6.0){
           out <- 4:5
  } else if( x >= 5.0){
    out <- 3:5
  } else if (x >= 4.0){
    out <- 2:5
  } else {
    out <- 1:5
  }
  return(out)
}

接下来，创建一个要删除的列索引列表：

z <- lapply(DF$C3, f)

最后，循环遍历每一行，并将相应的列设置为NA：

for(j in seq(length(z))){
  DF2[j, z[[j]]] <- NA
}

#-----
  V1 V2 V3 V4 V5
1  3  3 NA NA NA
2  2  7  2 NA NA
3 NA NA NA NA NA
4  4  5  7  3 NA
5  5 NA NA NA NA

- Chase

Answer 3

使用定义的切割点，对kohske的答案进行微调：

breaksx <- cut(DF$C3,c(0,3,4,5,6,7,Inf),labels=FALSE)
for (i in seq(nrow(DF2))) {
        DF2[i,breaksx[i]:ncol(DF2)] <- NA
}

结果：

> DF2
  V1 V2 V3 V4 V5
1  3  3 NA NA NA
2  2  7  2 NA NA
3 NA NA NA NA NA
4  4  5  7  3 NA
5  5 NA NA NA NA

删除所有值为NA的行

DF2[apply(DF2,1,function(x) !all(is.na(x))),]

结果：

  V1 V2 V3 V4 V5
1  3  3 NA NA NA
2  2  7  2 NA NA
4  4  5  7  3 NA
5  5 NA NA NA NA

- thelatemail