[英]R loop using data.table
我有一个大约 300 万行的数据集。 我创建了一个如下所示的小示例:
ex <- data.table(eoc = c(1,1,1,1,1,2,2,2,3,3), proc1 = c(63035,63020,92344,63035,27567,63020,1234,55678,61112,1236), trigger_cpt = c(63020,63020,63020,63020,63020,63020,63020,63020,61112,61112))
我有另一个 42 行的数据集,但生成了一个较小的示例:
add_on <- data.table(primary = c(63020,61112), secondary=c(63035,63445))
如果trigger_cpt 值恰好是数据集主列中的值之一,并且proc1 值是 add_on 中的辅助值,我需要重新标记“trigger_cpt”列(按 eoc 分组)上的某些行数据集。 如果符合条件,则应将 trigger_cpt 重新标记为辅助代码。
我首先手动输入所有内容, ex[,trigger_new:= if(any(trigger_cpt == '63020' & proc1 == '63035')) 63035 else trigger_cpt, eoc]
然后决定做一个for循环
for(i in 1:nrow(add_on)){
ex[,trigger_new2 := if(any(trigger_cpt == add_on[i,1] & proc1 == add_on[i,2])) add_on[i,2] else trigger_cpt, eoc]
}
但是,现在我正在我的 300 万行数据集上尝试此代码,运行它需要很长时间。 我不确定是否有更好的方法,或者是否可以对当前代码进行任何修改?
任何帮助将不胜感激!
预期 output:
ex_final <- data.table(eoc = c(1,1,1,1,1,2,2,2,3,3), proc1 = c(63035,63020,92344,63035,27567,63020,1234,55678,61112,1236), trigger_cpt = c(63035,63035,63035,63035,63035,63020,63020,63020,61112,61112))
这是一种产生 data.table 的方法,如果在分组集中找到匹配项,则将所有 trigger_cpt 设置为辅助值:
ex2 <- add_on[ex, , on=.(primary=trigger_cpt)][ , trigger_new := fifelse( secondary %in% proc1, secondary, NA_real_ ), by=eoc ]
ex.final <- ex2[ , trigger_cpt := fcoalesce( trigger_new, primary ) ][, .(eoc,proc1,trigger_cpt) ]
Output:
> ex.final
eoc proc1 trigger_cpt
1: 1 63035 63035
2: 1 63020 63035
3: 1 92344 63035
4: 1 63035 63035
5: 1 27567 63035
6: 2 63020 63020
7: 2 1234 63020
8: 2 55678 63020
9: 3 61112 61112
10: 3 1236 61112
此外,如果可行(这是有代价的),我会考虑使用setkey
,除非它弊大于利。 (初始处理可能使其不值得)。 它加快了下游操作,它可能使连接代码更清晰。 data.table 代码可能已经够难了。 因此:
setkey(ex, trigger_cpt )
setkey(add_on, primary )
## can now do this:
add_on[ex]
## instead of this:
add_on[ex, , on=.(primary=trigger_cpt)]
## .. in the code above.
... 此外...
如果您正在修改上述步骤,您会注意到add_on[ex]
(这是在 data.table 中进行左连接的有点倒退的方式),为您留下了add_on
的键列名称,而不是ex
。 这并不重要,只要您知道并最终适当地重命名列,但是加入数据的另一种方法可能是这样的:
ex2 <- merge( ex, add_on, by.x="trigger_cpt", by.y="primary" )
## and then work your way till the end with what this gives you
基于预期的 output
ex[, trigger_new := first(proc1), eoc]
ex
eoc proc1 trigger_cpt trigger_new
1: 1 63035 63020 63035
2: 1 63020 63020 63035
3: 1 92344 63020 63035
4: 1 63035 63020 63035
5: 1 27567 63020 63035
6: 2 63020 63020 63020
7: 2 1234 63020 63020
8: 2 55678 63020 63020
9: 3 61112 61112 61112
10: 3 1236 61112 61112
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.