R：為數據幀的每個子集查找斜率

Question

我有一個像這樣的數據：

dat <- data.frame(ID=sample(1:10, 100, rep=T),
                  Date=seq(as.Date("1982/01/01"), by="16 days", length.out = 100),
                  Value1=runif(100))

我需要按年份和ID來對數據進行子集，並為1月至6月和7月至12月的數據擬合一條線，並寫出2個斜率系數，我需要對年份和ID的所有組合進行此操作。

除了循環之外，還有其他方法嗎？ 實際數據有21788928行，循環花費的時間太長。

Answer 1

這應該更快，但是我不確定是否足夠快以滿足您的需求。 讓我知道：

library(dplyr)
library(lubridate)

# Function to return the coefficients of the regression as a data frame
coef.fcn = function(df) {
  coeffs = coef(lm(Value1 ~ Date, data=df))
  return(data.frame(Intercept=coeffs[1], Value1=coeffs[2]))
}

lm_coefs = dat %>% 
  mutate(my.cat = ifelse(month(Date) %in% 1:6, 
                         paste("Jan-Jun", year(Date)), paste("Jul-Dec", year(Date)))) %>%
  group_by(ID, my.cat) %>%
  do(coef.fcn(.))

這是部分示例數據結果：

lm.coefs

   ID       my.cat    Intercept        Value1
1   1 Jan-Jun 1983   0.62824396            NA
2   1 Jan-Jun 1985   0.71865235            NA
3   1 Jul-Dec 1985  20.20901291 -0.0033972977
4   2 Jan-Jun 1983 -37.54324401  0.0078885381
...
45  8 Jan-Jun 1982 -30.39203349  0.0068229828
46  8 Jan-Jun 1984 -27.62517465  0.0054096259
47  8 Jan-Jun 1985  27.70049296 -0.0048539844
48  8 Jul-Dec 1982  12.90814643 -0.0025997511
49  8 Jul-Dec 1984 -16.84585961  0.0032229997
...
57 10 Jan-Jun 1982   0.63533344            NA
58 10 Jan-Jun 1983   0.35107513            NA
59 10 Jan-Jun 1984   0.59588750            NA
60 10 Jul-Dec 1982   0.05156481            NA
61 10 Jul-Dec 1983   0.54658810            NA