Reputation: 3963

Lag multiple variables multiple times in R

So, I'm working with a data frame that has daily data over a period of 444 days. I have several variables that I want to lag for use in a regression model (lm). I want to lag them 7 times each. I'm currently generating the lags like this...

email_data$email_reach1 <- lag(ts(email_data$email_reach, start = 1, end = 444), 1)
email_data$email_reach2 <- lag(ts(email_data$email_reach, start = 1, end = 444), 2)
email_data$email_reach3 <- lag(ts(email_data$email_reach, start = 1, end = 444), 3)
email_data$email_reach4 <- lag(ts(email_data$email_reach, start = 1, end = 444), 4)
email_data$email_reach5 <- lag(ts(email_data$email_reach, start = 1, end = 444), 5)
email_data$email_reach6 <- lag(ts(email_data$email_reach, start = 1, end = 444), 6)
email_data$email_reach7 <- lag(ts(email_data$email_reach, start = 1, end = 444), 7)

Then, I repeat this for every single variable I want to lag.

This seems like a terrible way of accomplishing this. Is there something better?

I've thought about lagging the entire data frame, which works, but I don't know how you'd assign variable names to the result and merge it back to the original data frame.

Upvotes: 8

Answers (6)

Sebastian

Reputation: 1369

collapse::flag provides a general and fast (C++ based) solution to this problem:

library(collapse)
# Time-series (also supports xts and others)
head(flag(AirPassengers, -1:2))
##           F1  --  L1  L2
## Jan 1949 118 112  NA  NA
## Feb 1949 132 118 112  NA
## Mar 1949 129 132 118 112
## Apr 1949 121 129 132 118
## May 1949 135 121 129 132
## Jun 1949 148 135 121 129

# Time-series matrix
head(flag(EuStockMarkets, -1:2))
## Time Series:
## Start = c(1991, 130) 
## End = c(1998, 169) 
## Frequency = 260 
##           F1.DAX     DAX  L1.DAX  L2.DAX F1.SMI    SMI L1.SMI L2.SMI F1.CAC    CAC L1.CAC L2.CAC F1.FTSE   FTSE L1.FTSE L2.FTSE
## 1991.496 1613.63 1628.75      NA      NA 1688.5 1678.1     NA     NA 1750.5 1772.8     NA     NA  2460.2 2443.6      NA      NA
## 1991.500 1606.51 1613.63 1628.75      NA 1678.6 1688.5 1678.1     NA 1718.0 1750.5 1772.8     NA  2448.2 2460.2  2443.6      NA
## 1991.504 1621.04 1606.51 1613.63 1628.75 1684.1 1678.6 1688.5 1678.1 1708.1 1718.0 1750.5 1772.8  2470.4 2448.2  2460.2  2443.6
## 1991.508 1618.16 1621.04 1606.51 1613.63 1686.6 1684.1 1678.6 1688.5 1723.1 1708.1 1718.0 1750.5  2484.7 2470.4  2448.2  2460.2
## 1991.512 1610.61 1618.16 1621.04 1606.51 1671.6 1686.6 1684.1 1678.6 1714.3 1723.1 1708.1 1718.0  2466.8 2484.7  2470.4  2448.2
## 1991.515 1630.75 1610.61 1618.16 1621.04 1682.9 1671.6 1686.6 1684.1 1734.5 1714.3 1723.1 1708.1  2487.9 2466.8  2484.7  2470.4

# Data frame
head(flag(airquality[1:3], -1:2))
##   F1.Ozone Ozone L1.Ozone L2.Ozone F1.Solar.R Solar.R L1.Solar.R L2.Solar.R F1.Wind Wind L1.Wind L2.Wind
## 1       36    41       NA       NA        118     190         NA         NA     8.0  7.4      NA      NA
## 2       12    36       41       NA        149     118        190         NA    12.6  8.0     7.4      NA
## 3       18    12       36       41        313     149        118        190    11.5 12.6     8.0     7.4
## 4       NA    18       12       36         NA     313        149        118    14.3 11.5    12.6     8.0
## 5       28    NA       18       12         NA      NA        313        149    14.9 14.3    11.5    12.6
## 6       23    28       NA       18        299      NA         NA        313     8.6 14.9    14.3    11.5

# Panel lag
head(flag(iris[1:2], -1:2, iris$Species))
## Panel-lag computed without timevar: Assuming ordered data
##   F1.Sepal.Length Sepal.Length L1.Sepal.Length L2.Sepal.Length F1.Sepal.Width Sepal.Width L1.Sepal.Width L2.Sepal.Width
## 1             4.9          5.1              NA              NA            3.0         3.5             NA             NA
## 2             4.7          4.9             5.1              NA            3.2         3.0            3.5             NA
## 3             4.6          4.7             4.9             5.1            3.1         3.2            3.0            3.5
## 4             5.0          4.6             4.7             4.9            3.6         3.1            3.2            3.0
## 5             5.4          5.0             4.6             4.7            3.9         3.6            3.1            3.2
## 6             4.6          5.4             5.0             4.6            3.4         3.9            3.6            3.1

Similarly collapse::fdiff and collapse::fgrowth support stuitably lagged /leaded and iterated (quasi-, log-) differences and growth rates on (multivariate) time series and panels.

Upvotes: 1

John Chrysostom

Reputation: 3963

Based on the answer by Molx, but generalized for any list of variables, and patched up a bit... Thanks Molx!

do_lag <- function(the_data, variables, num_periods) {
  num_vars <- length(variables)
  num_rows <- nrow(the_data)

  for (j in 1:num_vars) {
    for (i in 1:num_periods) {
      the_data[[paste0(variables[j], i)]] <- c(rep(NA, i), head(the_data[[variables[j]]], num_rows - i))
    }
  }

  return(the_data)
}

Upvotes: 1

Khashaa

Reputation: 7373

You can also use data.table. (HT to @akrun)

set.seed(1)
email_data <- data.frame(dates=1:10, email_reach=rbinom(10, 10, 0.5))

library(data.table)
setDT(email_data)[, paste0('email_reach', 1:3) := shift(email_reach, 1:3)][]

#   dates email_reach email_reach1 email_reach2 email_reach3
# 1:     1           4           NA           NA           NA
# 2:     2           4            4           NA           NA
# 3:     3           5            4            4           NA
# 4:     4           7            5            4            4
# 5:     5           4            7            5            4
# 6:     6           7            4            7            5
# 7:     7           7            7            4            7
# 8:     8           6            7            7            4
# 9:     9           6            6            7            7
#10:    10           3            6            6            7

Upvotes: 8

IRTFM

Reputation: 263332

This isn't really an answer, just using the answer format as an elaboration of my warning above:

email_data <- data.frame( email_reach=ts(email_data$email_reach, start = 1, end = 444))

Then your code and this is what you get:

> head(email_data, 10)
   email_reach email_reach1 email_reach2 email_reach3 email_reach4
1            4            4            4            4            4
2            4            4            4            4            4
3            5            5            5            5            5
4            7            7            7            7            7
5            4            4            4            4            4
6            7            7            7            7            7
7            7            7            7            7            7
8            6            6            6            6            6
9            6            6            6            6            6
10           3            3            3            3            3
   email_reach5 email_reach6 email_reach7
1             4            4            4
2             4            4            4
3             5            5            5
4             7            7            7
5             4            4            4
6             7            7            7
7             7            7            7
8             6            6            6
9             6            6            6
10            3            3            3

Upvotes: 1

J.R.

Reputation: 3888

Another approach is to use the xts library. A little example follows, we start out with:

x <- ts(matrix(rnorm(100),ncol=2), start=c(2009, 1), frequency=12) 
head(x)
        Series 1   Series 2
[1,] -1.82934747 -0.1234372
[2,]  1.08371836  1.3365919
[3,]  0.95786815  0.0885484
[4,]  0.59301446 -0.6984993
[5,] -0.01094955 -0.3729762
[6,] -0.19256525  0.3137705

Convert it to xts, an call lag(), here with 0,1,2 lags to minimize output:

library(xts)
head(lag(as.xts(x),0:2))
            Series.1   Series.2  Series.1.1 Series.2.1 Series.1.2 Series.2.2
jan 2009 -1.82934747 -0.1234372          NA         NA         NA         NA
feb 2009  1.08371836  1.3365919 -1.82934747 -0.1234372         NA         NA
mar 2009  0.95786815  0.0885484  1.08371836  1.3365919 -1.8293475 -0.1234372
apr 2009  0.59301446 -0.6984993  0.95786815  0.0885484  1.0837184  1.3365919
maj 2009 -0.01094955 -0.3729762  0.59301446 -0.6984993  0.9578682  0.0885484
jun 2009 -0.19256525  0.3137705 -0.01094955 -0.3729762  0.5930145 -0.6984993

Upvotes: 4

Molx

Reputation: 6921

I think this does the same as your code above, for any given n.

n <- 7
for (i in 1:n) {
  email_data[[paste0("email_reach", i)]] <- lag(ts(email_data$email_reach, start = 1, end = 444), i)  
}

Upvotes: 2

Lag multiple variables multiple times in R

Answers (6)

Related Questions