在特定列中的特定值序列之后删除行

时间:2015-03-09 09:35:12

标签: r rows

a <- c("A","A","A","B","B","B","C","C","C","C","D","D","D","D","D")
b <- c("x","y","z","x","x","z","y","z","z","z","y","z","z","z","x")
df = data.frame(a,b)


    a   b
1   A   x
2   A   y
3   A   z
4   B   x
5   B   x
6   B   z
7   C   y
8   C   z
9   C   z
10  C   z
11  D   y
12  D   z
13  D   z
14  D   z
15  D   x

对于每个组A,B,C,D,我希望每次组合y,z出现在组的末尾时删除b列中的值z。

如果我们有一个==&#34; C&#34;的情况,其中b值是y,z,z,z,我想删除所有z&#39; s。但是,在==&#34; D&#34;中,没有必要改变,因为x是最后一个值。

结果如下:

    a   b
1   A   x
2   A   y
4   B   x
5   B   x
6   B   z
7   C   y
11  D   y
12  D   z
13  D   z
14  D   z
15  D   x

通过在dplyr中进行分组,我可以确定A中每个值的最后一次出现,因此a=="A"中描述的基本案例不是问题。我很难找到a=="C"的情况下的解决方案,在那里我可以发生一次y,然后发生20次z。

4 个答案:

答案 0 :(得分:4)

您可以在基础R中使用bycummin

df[unlist(by(df$b, interaction(df$a), FUN = function(x) {
  tmp <- rev(cummin(rev(x == "z")))
  if (tail(x[!tmp], 1) == "y") !tmp else rep(TRUE, length(x))
})), ]

结果:

   a b
1  A x
2  A y
4  B x
5  B x
6  B z
7  C y
11 D y
12 D z
13 D z
14 D z
15 D x

答案 1 :(得分:3)

这是一个可能的data.table解决方案。基本上,我正在创建一个满足3个条件的逻辑索引:成为z,第一个z出现在y之后,最后一个值为z然后我就是在评估它。

library(data.table)
setDT(df)[, indx := b == "z" & 
            max(which(b == "z")) == .N &
            ifelse(min(which(b == "z")) == 1L, 
                   TRUE, 
                   b[min(which(b == "z")) - 1L] == "y"),
            by = a][!(indx)]
#     a b  indx
#  1: A x FALSE
#  2: A y FALSE
#  3: B x FALSE
#  4: B x FALSE
#  5: B z FALSE
#  6: C y FALSE
#  7: D y FALSE
#  8: D z FALSE
#  9: D z FALSE
# 10: D z FALSE
# 11: D x FALSE

答案 2 :(得分:1)

这是一个基本解决方案:

do.call("rbind", by(df, df$a, FUN = function(x) {
    if(x$b[length(x$b)] == "z") {
        y <- which(x$b == "y")
        if(!length(y)) {
            return(x)
        }
        z <- which(x$b == "z")
        if(!length(z)) {
            return(x)
        }
        # check if y isn't immediately before z
        if(max(y) - min(z) > 1) {
            return(x)
        } else {
            return(x[-z,])
        }
    } else {
        return(x)
    }
}))

结果:

     a b
A.1  A x
A.2  A y
B.4  B x
B.5  B x
B.6  B z
C    C y
D.11 D y
D.12 D z
D.13 D z
D.14 D z
D.15 D x

答案 3 :(得分:1)

效率不高,但效果很好:

require(stringr)
df2 <- data.frame(row.names = c("a", "b")) 

for(i in levels(factor(df$a))) {
  temp <- paste(df$b[df$a == i], collapse = "")
  if(str_detect(temp, "yz") & str_detect(temp, "z$")) {
    temp <- gsub("z", "", temp)
    df2 <- rbind(df2, data.frame(a = rep(i, nchar(temp)), b = substring(temp, seq(1,nchar(temp),1), seq(1,nchar(temp),1))))
  } else df2 <- rbind(df2, data.frame(a = rep(i, nchar(temp)), b = substring(temp, seq(1,nchar(temp),1), seq(1,nchar(temp),1))))   
}

#    a b
# 1  A x
# 2  A y
# 3  B x
# 4  B x
# 5  B z
# 6  C y
# 7  D y
# 8  D z
# 9  D z
# 10 D z
# 11 D x