基于 r 中的另一个变量生成标记变量

Question

我正在尝试生成一个标记变量。这是我的示例数据集的样子：

data <- data.frame(
  ids = c(1,1,1, 2,2,2, 3,3,3),
  cat = c("A","B","B", "C","B","C", "A","B","A"))

> data
  ids cat
1   1   A
2   1   B
3   1   B
4   2   C
5   2   B
6   2   C
7   3   A
8   3   B
9   3   A

如果每个 id.

没有观察到 C，我需要一个标记 ids 的标记变量

 > data
      ids cat  flag
    1   1   A    0
    2   1   B    0
    3   1   B    0
    4   2   C    1
    5   2   B    1
    6   2   C    1
    7   3   A    0
    8   3   B    0
    9   3   A    0

有什么想法吗？谢谢！

Answer 1

按 'ids' 分组后，通过检查 'C' 是否为 %in% 'cat'、returns 的逻辑值来创建 'flag' TRUE/FALSE，用 as.integer 或 +

强制为 1/0

library(dplyr)
data %>%
     group_by(ids) %>%
     mutate(flag = +('C' %in% cat))
# A tibble: 9 x 3
# Groups:   ids [3]
#    ids cat    flag
#  <dbl> <fct> <int>
#1     1 A         0
#2     1 B         0
#3     1 B         0
#4     2 C         1
#5     2 B         1
#6     2 C         1
#7     3 A         0
#8     3 B         0
#9     3 A         0

或用any

换行

  ... %>%
  mutate(flag = as.integer(any(cat %in% 'C')))

基于 r 中的另一个变量生成标记变量

Generating a flagging variable based on another variable in r

r

recode