r - 在 [R] 中，为组的每个值生成新变量

Question

我有 id 变量和 date 变量，其中给定 id （面板）有多个日期。我想根据给定 id 的任何年份是否满足逻辑条件来生成一个新变量。我不确定如何对其进行编码，因此请不要将以下内容视为 R 代码，就像逻辑伪代码一样。就像是

foreach(i in min(id):max(id)) {
if(var1[yearvar[1:max(yearvar)]=="A") then { newvar==1}
}

举个例子：

ID     Year     Letter
1     1999        A
1     2000        B
2     2000        C
3     1999        A

应该返回newvar 1 1 0 1

既然data[ID==1]在某年包含A，那么==1尽管那一年它也应该在2000Letter==B年。

Answer 1

这是使用的解决方案plyr：

library(plyr)
a <- ddply(dat, .(ID), summarise, newvar = as.numeric(any(Letter == "A")))
merge(ID, a, by="ID")

Answer 2

这是一种使用基数 R 接近它的方法：

#Find which ID meet first criteria
withA <- unique(dat$ID[dat$Letter == "A"])

#add new column based on whether ID is in withA
dat$newvar <- as.numeric(dat$ID %in% withA)

#    ID Year Letter newvar
# 1  1 1999      A      1
# 2  1 2000      B      1
# 3  2 2000      C      0
# 4  3 1999      A      1

Answer 3

不使用包：

dat <- data.frame(
    ID = c(1,1,2,3),
    Year = c(1999,2000,2000,1999),
    Letter = c("A","B","C","A")
)
tableData <- table(dat[,c("ID","Letter")])
newvar <- ifelse(tableData[dat$ID,"A"]==1,1,0)
dat <- cbind(dat,newvar)

#  ID Year Letter newvar
#1  1 1999      A      1
#2  1 2000      B      1
#3  2 2000      C      0
#4  3 1999      A      1