时间戳

首页 > 解决方案 > 根据少数条件创建新变量

r - 根据少数条件创建新变量

问题描述

在附图中，我想在 R 中创建两个组：如果任何 ID 的一半天数 >=90，则在新类别中返回 A，否则，在新类别中返回 B。我还想保留/返回所有其他变量。

在此处输入图像描述

抱歉，我对 R 很陌生。

标签： rif-statementconditional-statements

解决方案

您可以使用 package.json 尝试以下dplyr操作。分组后ID，new_category通过将大于或等于 90 的实例相加days，然后查看是否大于该行的“一半”来创建ID。如果是，则设置new_category为“A”（否则为“B”）。

set.seed(123)

library(dplyr)

df <- data.frame(
  visit_date = Sys.Date() + sample(1:100, 9),
  days = c(90, 30, 30, 120, 90, 20, 90, 90, 30),
  ID = c(1,1,1,2,2,3,3,3,3)
)

df %>%
  group_by(ID) %>%
  mutate(new_category = ifelse(
    sum(days >= 90) >= n()/2,
    "A",
    "B"
  ))

这是与基础 R 类似的方法：

df$new_category = ave(df$days, 
                      df$ID, 
                      FUN = function(x) ifelse(
                        sum(x >= 90) >= length(x) / 2, 
                        "A", 
                        "B")
                      )

输出

  visit_date days ID new_category
1 2021-10-26   90  1            B
2 2021-12-13   30  1            B
3 2021-11-15   30  1            B
4 2021-10-09  120  2            A
5 2021-12-01   90  2            A
6 2021-11-06   20  3            A
7 2021-11-14   90  3            A
8 2021-11-07   90  3            A
9 2021-12-31   30  3            A

数据

df <- structure(list(visit_date = structure(c(18926, 18974, 18946, 
18909, 18962, 18937, 18945, 18938, 18992), class = "Date"), days = c(90, 
30, 30, 120, 90, 20, 90, 90, 30), ID = c(1, 1, 1, 2, 2, 3, 3, 
3, 3), new_category = c("B", "B", "B", "A", "A", "A", "A", "A", 
"A")), row.names = c(NA, -9L), class = "data.frame")

推荐阅读