有条件地更新 data.table 中的单元格

Question

我有一个季度观察数据表，我想将其与医疗补助扩展相关联。为此，我希望我的 data.table (expanded_medicaid) 中有一个变量，如果状态扩展它，则为 1，否则为 0。我有另一个 data.table，其中包含扩展日期。

所以，例如，我的主要 data.table 是这样的：

state_name  qtr     ...data columns
Alabama     2008 Q1  
Alabama     2008 Q2
...
Alaska      2008 Q1
Alaska      2008 Q2
... etc

另一个数据表有扩展日期

state_name    qtr
Alaska        2015 Q1
Arizona       2014 Q1
Arkansas      2014 Q1
... etc

如果 qtr >= 第二个 data.table 中该州的 qtr，我是否可以创建一个列 expanded_medicaid 为 1？ （这似乎是 sapply 的工作，但我从来没有正确理解 sapply 的工作原理。）

编辑：理想的结果是主表如下所示：

state_name   qtr      expanded_medicaid   ...data columns
...
Alaska       2013 Q3  0
Alaska       2013 Q4  0
Alaska       2014 Q1  0
Alaska       2014 Q2  0
Alaska       2014 Q3  0
Alaska       2014 Q4  0
Alaska       2015 Q1  1
Alaska       2015 Q2  1
...
Arizona      2013 Q3  0
Arizona      2013 Q4  0
Arizona      2014 Q1  1
Arizona      2014 Q2  1
Arizona      2014 Q3  1
Arizona      2014 Q4  1
Arizona      2015 Q1  1
Arizona      2015 Q2  1
...

Answer 1

您可以将data.table 与zoo 的as.yearqtr() 函数结合起来比较季度

library(zoo)
library(data.table)

df1 <- data.table(state_name = c('Alabama', 'Arizona', 'Arkanas'),
                  qtr = c('2008 Q1', '2009 Q2', '2009 Q3'))

df2 <- data.table(state_name = c('Alabama', 'Arizona', 'Arkanas'),
                  qtr = c('2007 Q1', '2012 Q2', '2009 Q3'))

df1[, qtr := as.yearqtr(qtr, format = "%Y Q%q")]
df2[, qtr := as.yearqtr(qtr, format = "%Y Q%q")]

df1[df2, on = .(state_name), expanded_medicaid := as.integer(qtr >= i.qtr)]

#   state_name     qtr expanded_medicaid
#1:    Alabama 2008 Q1                 1
#2:    Arizona 2009 Q2                 0
#3:    Arkanas 2009 Q3                 1