仅当 R 中 2 个变量的观察值彼此匹配时才子集数据

Question

我有这样的数据：

a <- c("blue", "red", "green", "blue","cyan")
b <- c("red","red","green","blue", "orange")
df <- data.frame(a,b)
df

      a      b
1  blue    red
2   red    red
3 green  green
4  blue   blue
5  cyan orange

如果蓝色和红色相互匹配，我正在尝试对行进行子集化并自己观察。

我正在尝试以下代码，但是当我看到使用表函数检查它们时，仍有一些其他颜色与其中一种颜色匹配。

 sub <- df[df$a %in% c("blue", "red" & df$b %in% c("blue","red"), ]
 table(sub$a, sub$b)

这对我来说非常棘手。只有当蓝色和红色相互匹配并自己观看时，我如何才能告诉 R 子集？

期望的输出是：

      a      b
1  blue    red
2   red    red
3  blue   blue

这样做的最终目标是通过从 5 x 5 列联表中分离出来来创建 2 x 2 列联表。如果有其他建议可以做到这一点，我们将不胜感激。

提前致谢！

这就是我不想要的意思。我只想保持观察蓝色和红色观察。我不想观察绿色、橙色、青色。

            Blue        Red            Green        Orange   Cyan
  Blue       28          39              32            3        1  
  Red        47         244             184           56        3
  Green      0           0               0            0         0
  Orange     0           0               0            0         0
  Cyan       0           0               0            0         0

Answer 1

您可以添加droplevels() 函数，如下：

# here the markus solution
twobytwo <- df[which(df$a %in% c("blue", "red") & df$b %in% c("blue","red")), ]
#here the droplevels, that removes the unused level
table(droplevels(twobytwo))

         b
a      blue red
  blue    1   1
  red     0   1

更多信息here.

Answer 2

这应该可行！

output <- df[df$a %in% c('red','blue') & df$b %in% c('red','blue'),]

Answer 3

您可以尝试使用grepl 过滤您的data.frame：

require(tidyverse)

result <- df %>% 
  varhandle::unfactor() %>%
  filter(grepl(pattern = paste(c("red", "blue"), collapse="|"), a) |
         grepl(pattern = paste(c("red", "blue"), collapse="|"), b))

result
     a    b
1 blue  red
2  red  red
3 blue blue

table(result)
      b
a      blue red
  blue    1   1
  red     0   1