根据 r 中的另一个变量生成标记变量答案

【问题标题】：Generating a flagging variable based on another variable in r根据 r 中的另一个变量生成标记变量
【发布时间】：2020-02-11 22:02:34
【问题描述】：

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

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

有什么想法吗？谢谢！

【问题讨论】：

这是一个基本的 R 解决方案：data$flag <- as.integer(data$cat %in% "C")

标签： r recode

【解决方案1】：

按'ids'分组后，通过检查'C'是否为%in%'cat'来创建'flag'，返回逻辑值TRUE/FALSE，用as.integer或@987654323强制为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')))

【讨论】：