как преобразовать вложенный l oop в lapply в r - PullRequest
Новая
       4

как преобразовать вложенный l oop в lapply в r

2 голосов
/ 08 февраля 2020

У меня есть список с именем «dahak», который содержит 30000 чисел от 1 до 10. Я хочу проверить каждое число со всеми числами в списке, если два числа равны, тогда добавьте число 1 в weight_list, если два числа не равны, то рассчитайте их разницу и сохраните ее как x и добавьте x в weight_list. Вот код: 

for(j in 1:num_nodes){
  for (k in 1:num_nodes){
    if(j==k){
      weight_list <- c(weight_list,0)
    }
    else if(as.numeric(dahak[j])==as.numeric(dahak[k])){
      weight_list <- c(weight_list,1)
    }
    else if(as.numeric(dahak[j])!=as.numeric(dahak[k])){
      x = 1 - (abs(as.numeric(dahak[j]) - as.numeric(dahak[k])) / 10)
      weight_list <- c(weight_list,x)
    }
  }
}

Как я могу оптимизировать этот код? и как я могу сделать это с lapply?

Ответы [ 2 ]

3 голосов
/ 08 февраля 2020

Звучит так, будто вы хотите создать матрицу 30 000 x 30 000. Звучит также, что dahak - это вектор, а не список. Если это действительно , что вы хотите сделать, вы можете упростить свою логику c и векторизовать так: 

get_weights <- function(x)  1 - abs(x - as.numeric(dahak))/10 
weights <- do.call(rbind, lapply(as.numeric(dahak), get_weights)) - diag(length(dahak))

Используя те же фиктивные данные, что и @ThomasIsCoding, я получаю: 

weights
#>       [,1] [,2] [,3] [,4] [,5] [,6] [,7] [,8] [,9] [,10] [,11] [,12] [,13] [,14] [,15]
#>  [1,]  0.0  0.9  0.7  0.3  1.0  0.4  0.3  0.6  0.6   0.8   1.0   0.9   0.6   0.9   0.5
#>  [2,]  0.9  0.0  0.8  0.4  0.9  0.5  0.4  0.7  0.7   0.7   0.9   0.8   0.7   1.0   0.6
#>  [3,]  0.7  0.8  0.0  0.6  0.7  0.7  0.6  0.9  0.9   0.5   0.7   0.6   0.9   0.8   0.8
#>  [4,]  0.3  0.4  0.6  0.0  0.3  0.9  1.0  0.7  0.7   0.1   0.3   0.2   0.7   0.4   0.8
#>  [5,]  1.0  0.9  0.7  0.3  0.0  0.4  0.3  0.6  0.6   0.8   1.0   0.9   0.6   0.9   0.5
#>  [6,]  0.4  0.5  0.7  0.9  0.4  0.0  0.9  0.8  0.8   0.2   0.4   0.3   0.8   0.5   0.9
#>  [7,]  0.3  0.4  0.6  1.0  0.3  0.9  0.0  0.7  0.7   0.1   0.3   0.2   0.7   0.4   0.8
#>  [8,]  0.6  0.7  0.9  0.7  0.6  0.8  0.7  0.0  1.0   0.4   0.6   0.5   1.0   0.7   0.9
#>  [9,]  0.6  0.7  0.9  0.7  0.6  0.8  0.7  1.0  0.0   0.4   0.6   0.5   1.0   0.7   0.9
#> [10,]  0.8  0.7  0.5  0.1  0.8  0.2  0.1  0.4  0.4   0.0   0.8   0.9   0.4   0.7   0.3
#> [11,]  1.0  0.9  0.7  0.3  1.0  0.4  0.3  0.6  0.6   0.8   0.0   0.9   0.6   0.9   0.5
#> [12,]  0.9  0.8  0.6  0.2  0.9  0.3  0.2  0.5  0.5   0.9   0.9   0.0   0.5   0.8   0.4
#> [13,]  0.6  0.7  0.9  0.7  0.6  0.8  0.7  1.0  1.0   0.4   0.6   0.5   0.0   0.7   0.9
#> [14,]  0.9  1.0  0.8  0.4  0.9  0.5  0.4  0.7  0.7   0.7   0.9   0.8   0.7   0.0   0.6
#> [15,]  0.5  0.6  0.8  0.8  0.5  0.9  0.8  0.9  0.9   0.3   0.5   0.4   0.9   0.6   0.0
0 голосов
/ 08 февраля 2020

Я думаю, это может быть упрощение, которое вы ищете, где используются outer и ifelse.

Ниже приведен пример с фиктивными данными: 

set.seed(1)
num_nodes <- 15
dahak <- sample(10,num_nodes,replace = TRUE)

Если вам нужна матрица для weigth_list измерений num_nodes, тогда вы можете попробовать 

weight_list <- (u<-ifelse((z<-abs(outer(dahak,dahak,FUN = "-")))!=0,1-z/10,1))-diag(diag(u))

такой, что 

> weight_list
      [,1] [,2] [,3] [,4] [,5] [,6] [,7] [,8] [,9] [,10] [,11] [,12] [,13] [,14] [,15]
 [1,]  0.0  0.5  0.8  0.2  0.3  0.8  0.3  0.4  0.2   0.6   0.6   0.9   0.7   0.9   0.8
 [2,]  0.5  0.0  0.7  0.7  0.8  0.7  0.8  0.9  0.7   0.9   0.9   0.4   0.8   0.4   0.7
 [3,]  0.8  0.7  0.0  0.4  0.5  1.0  0.5  0.6  0.4   0.8   0.8   0.7   0.9   0.7   1.0
 [4,]  0.2  0.7  0.4  0.0  0.9  0.4  0.9  0.8  1.0   0.6   0.6   0.1   0.5   0.1   0.4
 [5,]  0.3  0.8  0.5  0.9  0.0  0.5  1.0  0.9  0.9   0.7   0.7   0.2   0.6   0.2   0.5
 [6,]  0.8  0.7  1.0  0.4  0.5  0.0  0.5  0.6  0.4   0.8   0.8   0.7   0.9   0.7   1.0
 [7,]  0.3  0.8  0.5  0.9  1.0  0.5  0.0  0.9  0.9   0.7   0.7   0.2   0.6   0.2   0.5
 [8,]  0.4  0.9  0.6  0.8  0.9  0.6  0.9  0.0  0.8   0.8   0.8   0.3   0.7   0.3   0.6
 [9,]  0.2  0.7  0.4  1.0  0.9  0.4  0.9  0.8  0.0   0.6   0.6   0.1   0.5   0.1   0.4
[10,]  0.6  0.9  0.8  0.6  0.7  0.8  0.7  0.8  0.6   0.0   1.0   0.5   0.9   0.5   0.8
[11,]  0.6  0.9  0.8  0.6  0.7  0.8  0.7  0.8  0.6   1.0   0.0   0.5   0.9   0.5   0.8
[12,]  0.9  0.4  0.7  0.1  0.2  0.7  0.2  0.3  0.1   0.5   0.5   0.0   0.6   1.0   0.7
[13,]  0.7  0.8  0.9  0.5  0.6  0.9  0.6  0.7  0.5   0.9   0.9   0.6   0.0   0.6   0.9
[14,]  0.9  0.4  0.7  0.1  0.2  0.7  0.2  0.3  0.1   0.5   0.5   1.0   0.6   0.0   0.7
[15,]  0.8  0.7  1.0  0.4  0.5  1.0  0.5  0.6  0.4   0.8   0.8   0.7   0.9   0.7   0.0
...