R 当不存在足够的行时，在 data.table 中按组查找运行相关性

问题描述

我正在尝试使用 runcor 按组查找运行相关性。由于最后一个团队只有一个元素，runcor 函数会抛出错误。

library(data.table)
library(TTR)
dat <- data.table(
  Team = sapply(1:32,function(x) paste0("T",x)),Year = c(rep(c(2000:2009),32)),Points_Game = c(rnorm(320,100,10))
)

dat = dat[order(Team,Year)][1:311]
# find correlation of Year and Points_Game for each Team
dat = dat[,r := TTR::runcor(Year,Points_Game,n = 3),by = Team]

有没有办法在找不到相关性的情况下捕获此案例（T9 团队）并在 T9 的 r 列中填充 NA？

谢谢！

解决方法

如果出现错误，您可以使用 tryCatch 返回 NA：

runCorProtected <- function(...) {
  tryCatch(TTR::runCor(...),error = function(e) {NA})
}

dat = dat[,r := runCorProtected(Year,Points_Game,n = 3),by = Team]

dat[Team=='T9']
   Team Year Points_Game  r
1:   T9 2000    103.1672 NA