Using rollapply to output to lists of lists

2019-08-02 11:05发布

问题:

I would like to use rollapply or rollapplyr to apply the modwt function to my time series data.

I'm familiar with how rollapply/r works but I need some help setting up the output so that I can correctly store my results when using rollapply.

The modwt function in the waveslim package takes a time series and decomposes it into J levels, for my particular problem J = 4 which means I will have 4 sets of coefficients from my single time series stored in a list of 5. Of this list I am only concerned with d1,d2,d3 & d4.

The output of the modwt function looks as follows

    > str(ar1.modwt)
List of 5
 $ d1: num [1:200] -0.223 -0.12 0.438 -0.275 0.21 ...
 $ d2: num [1:200] 0.1848 -0.4699 -1.183 -0.9698 -0.0937 ...
 $ d3: num [1:200] 0.5912 0.6997 0.5416 0.0742 -0.4989 ...
 $ d4: num [1:200] 1.78 1.86 1.85 1.78 1.65 ...
 $ s4: num [1:200] 4.64 4.42 4.19 3.94 3.71 ...
 - attr(*, "class")= chr "modwt"
 - attr(*, "wavelet")= chr "la8"
 - attr(*, "boundary")= chr "periodic"

In the example above I have applied the modwt function to the full length time series of length 200 but I wish to apply it to a small rolling window of 30 using rollapply.

I have already tried the following but the output is a large matrix and I cannot easily identify which values belong to d1,d2,d3 or d4

roller <- rollapplyr(ar1, 30,FUN=modwt,wf="la8",n.levels=4,boundary="periodic")

The output of this is a large matrix with the following structure:

> str(roller)
List of 855
 $ : num [1:30] 0.117 -0.138 0.199 -1.267 1.872 ...
 $ : num [1:30] -0.171 0.453 -0.504 -0.189 0.849 ...
 $ : num [1:30] 0.438 -0.3868 0.1618 -0.0973 -0.0247 ...
 $ : num [1:30] -0.418 0.407 0.639 -2.013 1.349 ...

...lots of rows omitted...

$ : num [1:30] 0.307 -0.658 -0.105 1.128 -0.978 ...
  [list output truncated]
 - attr(*, "dim")= int [1:2] 171 5
 - attr(*, "dimnames")=List of 2
  ..$ : NULL
  ..$ : chr [1:5] "d1" "d2" "d3" "d4" ...

How can I set up a variable such that it will store the (200-30)+1 lists with lists within this for each of the scales d1,d2,d3 and d4?

For a reproducible example please use the following:

library(waveslim)
data(ar1)
ar1.modwt <- modwt(ar1, "la8", 4)

回答1:

Define modwt2 which invokes modwt, takes the first 4 components and strings them out into a numeric vector. Then use rollapplyr with that giving rollr where each row of rollr is the result of one call to modwt2. Finally, reshape each row of rollr into a separate matrix and create a list, L, of those matrices:

modwt2 <- function(...) unlist(head(modwt(...), 4))
rollr <- rollapplyr(ar1, 30, FUN = modwt2, wf = "la8", n.levels = 4, boundary = "periodic")
L <- lapply(1:nrow(rollr), function(i) matrix(rollr[i,], , 4))

If a 30 x 4 x 171 array is desired then the following will simplify it into a 3d array:

simplify2array(L)

or as a list of lists:

lapply(L, function(x) as.list(as.data.frame(x)))

2) This is an alternate solution that just uses lapply directly and returns a list each of whose components is the list consisting of d1, d2, d3 and d4.

lapply(1:(200-30+1), function(i, ...) head(modwt(ar1[seq(i, length = 30)], ...), 4),
  wf = "la8", n.levels = 4, boundary = "periodic")

Updates: Code improvements, expand (1) and add (2).