2017-05-18 40 views
0

我有一個很大的矩陣/數據框(2500x20),並且需要一列滾動偏差,最大值小於所有先前單元的最小值 - 不包括當前值。R:矩陣上的絕對滾動偏差(帶偏移1)

我按列應用我的功能。我的函數將整列(作爲向量)移動1:length,產生一個列表列表。我需要這個矩陣來應用max(x)-min(x)函數。這適用於一個小矩陣,並且對於我需要的尺寸而言運行時間過長。

(小)源(提供):

 [,1] [,2] [,3] [,4] [,5] 
[1,] 55 9 99 0 NA 
[2,] 54 7 98 1 NA 
[3,] 56 12 97 2 NA 
[4,] 53 8 96 3 1 
[5,] 57 22 95 4 0 
[6,] 52 51 94 5 -1 
[7,] 58 6 93 6 NA 
[8,] 51 6 93 7 0 
[9,] 59 51 92 8 2 
[10,] 50 78 91 9 NA 
[11,] 60 12 90 10 NA 
[12,] 49 5 89 11 -2 

預期成果:

 [,1] [,2] [,3] [,4] [,5] 
[1,] 0 0 0 0 0 
[2,] 0 0 0 0 0 
[3,] 1 2 1 1 0 
[4,] 2 5 2 2 0 
[5,] 3 5 3 3 0 
[6,] 4 15 4 4 1 
[7,] 5 44 5 5 2 
[8,] 6 45 6 6 2 
[9,] 7 45 6 7 2 
[10,] 8 45 7 8 3 
[11,] 9 72 8 9 3 
[12,] 10 72 9 10 3 

第一個結果列將始終爲0,因爲它沒有以前的行可能從計算。
第二個結果行將始終爲零,因爲前一行是最大值和最小值。
最後的源代碼行永遠不會影響結果。

我有什麼至今:

library(data.table) 

mytest <- matrix(c(
55, 9,99, 0,NA, 
54, 7,98, 1,NA, 
56,12,97, 2,NA, 
53, 8,96, 3, 1, 
57,22,95, 4, 0, 
52,51,94, 5,-1, 
58, 6,93, 6,NA, 
51, 6,93, 7, 0, 
59,51,92, 8, 2, 
50,78,91, 9,NA, 
60,12,90,10,NA, 
49, 5,89,11,-2 
), ncol=5, byrow = TRUE) 

rolling_deviation <- function (a_column){ 
    tmp1 = shift(a_column, 1:(length(a_column)-1), NA, "lag") 
    tmp2 = matrix(unlist(tmp1), ncol = length(a_column), byrow = TRUE) 
    apply(tmp2,2,function(x){ 
     x = x[!is.na(x)] 
     ifelse(length(x)==0, 0, max(x) - min(x)) 
    }) 
} 

apply(mytest,2,rolling_deviation) 

我需要這個來計算速度更快,會有更多的行爲的觀察增加。

+0

您可能需要指定使用的軟件包 – akrun

+0

對不起,添加了庫(data.table)的代碼 - 我忽略了轉移。 –

回答

0

這個怎麼樣?

> system.time(outcome<- apply(mytest,2,rolling_deviation)) 
    user system elapsed 
    0.014 0.002 0.038 

> system.time(outcome1<- setDT(data.frame(mytest))[, lapply(.SD, rolling_deviation)]) 
    user system elapsed 
    0.002 0.000 0.002 

的結果是一樣的:

> outcome 
     [,1] [,2] [,3] [,4] [,5] 
[1,] 0 0 0 0 0 
[2,] 0 0 0 0 0 
[3,] 1 2 1 1 0 
[4,] 2 5 2 2 0 
[5,] 3 5 3 3 0 
[6,] 4 15 4 4 1 
[7,] 5 44 5 5 2 
[8,] 6 45 6 6 2 
[9,] 7 45 6 7 2 
[10,] 8 45 7 8 3 
[11,] 9 72 8 9 3 
[12,] 10 72 9 10 3 

> outcome1 
    X1 X2 X3 X4 X5 
1: 0 0 0 0 0 
2: 0 0 0 0 0 
3: 1 2 1 1 0 
4: 2 5 2 2 0 
5: 3 5 3 3 0 
6: 4 15 4 4 1 
7: 5 44 5 5 2 
8: 6 45 6 6 2 
9: 7 45 6 7 2 
10: 8 45 7 8 3 
11: 9 72 8 9 3 
12: 10 72 9 10 3 
+0

不幸的是,這對2500x20數據沒有影響 –

0

我通過創建一個嵌套循環解決我的問題。我被教導說,採用循環對性能幾乎總是不好,但我找不到更好的方法。我需要一個輔助函數來執行最小/最大操作,因爲NA參數很可能並不是我的目的錯誤。

rolling_range <- function(in_mat){ 
    ignore_na = function(op, a,b){ 
     if(is.na(a)){ 
      return(b) 
     }else if(is.na(b)){ 
      return(a) 
     }else{ 
      return(op(a,b)) 
     } 
    } 
    out_mat.min = matrix(NA, nrow = nrow(in_mat), ncol = ncol(in_mat)) 
    out_mat.max = matrix(NA, nrow = nrow(in_mat), ncol = ncol(in_mat)) 
    out_mat.result = matrix(0, nrow = nrow(in_mat), ncol = ncol(in_mat)) 
    for(m in 1:ncol(in_mat)){ 
     for(n in 2:nrow(in_mat)){ 
      out_mat.min[n,m] = ignore_na(`min`, out_mat.min[(n-1),m], in_mat[(n-1),m]) 
      out_mat.max[n,m] = ignore_na(`max`, out_mat.max[(n-1),m], in_mat[(n-1),m]) 
      out_mat.result[n,m] = ifelse(is.na(out_mat.min[n,m]) || is.na(out_mat.max[n,m]), 0, out_mat.max[n,m] - out_mat.min[n,m]) 
     } 
    } 
    return(out_mat.result) 
} 

rolling_range(mytest)