如何在 R 中仅保留满足特定条件的行

Question

我有一个数据框，其中包含大约 700 个案例和 1800 次检查。 有些病例经历了几种不同的方式。 我想根据模态的具体情况只留下一个检查结果。

这是一个虚拟数据框：

df <- data.frame (ID = c("1", "1", "1", "2", "2", "3", "4", "4", "5", "5"),
                  c1 = c("A", "B", "C", "A", "C", "A", "A", "B", "B", "C"),
                  x1 = c(5, 4, 5, 3, 1, 3, 4, 2, 3, 5),
                  x2 = c(4, 3, 7, 9, 1, 2, 4, 7, 5, 0))

有5个案例，10个考试。 [c1]是考试形式（条件），结果是x1和x2。

我想根据以下条件只留下一行：

C > B > A

我想先离开 C 行； 如果没有，则离开 B 行； 如果 C 和 B 不存在，则保留有 A 的行。

所需的 output：

output <- data.frame (ID = c("1", "2", "3", "4", "5"),
                      c1 = c("C", "C", "A", "B", "C"),
                      x1 = c(5, 1, 3, 2, 5),
                      x2 = c(7, 1, 2, 7, 0))

Answer 1

这是一种方法：

df.srt <- df[order(df$c1, decreasing=TRUE), ]
df.spl <- split(df.srt, df.srt$ID)
first <- lapply(df.spl, head, n=1)
result <- do.call(rbind, first)
result
#   ID c1 x1 x2
# 1  1  C  5  7
# 2  2  C  1  1
# 3  3  A  3  2
# 4  4  B  2  7
# 5  5  C  5  0

Answer 2

您可以根据所需的正确顺序arrange数据，对于每个ID select，它位于第一行。

library(dplyr)

req_order <- c('C', 'B', 'A')

df %>%
  arrange(ID, match(c1, req_order)) %>%
  distinct(ID, .keep_all = TRUE)

#  ID    c1       x1    x2
#  <chr> <chr> <dbl> <dbl>
#1 1     C         5     7
#2 2     C         1     1
#3 3     A         3     2
#4 4     B         2     7
#5 5     C         5     0

在基础 R 中，可以写成：

df1 <- df[order(match(df$c1, req_order)), ]
df1[!duplicated(df1$ID), ]