Управление подматрицами в R - PullRequest
Новая
       40

Управление подматрицами в R

0 голосов
/ 14 мая 2018 
Nh<-matrix(c(17,26,30,17,23, 17 ,24, 23), nrow=2, ncol=4); Nh
Sh<-matrix(c(8.290133, 6.241174, 6.096808, 7.4449672, 6.894924, 7.692115, 
             4.540521, 7.409122), nrow=2, ncol=4); Sh
NhSh<-as.matrix(Nh*Sh); NhSh
rh<-c( 0.70710678, 0.40824829, 0.28867513, 0.22360680, 0.18257419, 
       0.15430335, 0.13363062, 0.11785113, 0.10540926, 0.09534626); rh

pv <- c()
for (j in 1:2) {
  for (i in 1:4) {

    pv <- rbind(pv, NhSh[j,i]*rh)
  }
}
pv

row.names(pv) <- rep(c(1:2), each = 4)
lst<-lapply(split(seq_len(nrow(pv)), as.numeric(row.names(pv))), function(i) 
pv[i,])

data<-40
nlargest <- function(x, data) 
{
  res <- order(x)[seq_len(data)];
  pos <- arrayInd(res, dim(x), useNames = TRUE);
  list(values = pv[res], position = pos)
}
out <- lapply(lst, nlargest, data = 40)

В продолжение вышеприведенного кода. Есть ли какой-нибудь краткий способ повторить следующие шаги для каждой позиции $ ’k’ $ для k в 1: 2? 

s1<-c(1,1,1,1); ch<-c(5,7,10,5); C<-150; a<-out$'1'$position 
for (j in a[40:1, "row"] ) 
{
 s1[j] <- s1[j]+1;
 cost1 <- sum(ch*s1);
 if (cost1>=C) break
}
s1; cost1
#Output [1] 5 6 6 5  

#       [1] 152

Я должен получить 2 значения для 's' и 'стоимость' для нашей позиции $ k $. Я пытался 

mat = replicate (2,{x = matrix(data = rep(NA, 80), ncol = 2)}); mat
for (k in 1:2)
{
  mat[,,k]<-out$'k'$position
}
mat

Ошибка в mat [,, k] <- out $ k $ position: количество заменяемых элементов не кратно длине замены </p> 

for (k in 1:2)
{
for (j in mat[,,k][40:1] ) {
  s[j] <- s[j]+1  
  cost <- sum(ch*s)
  if (cost>=C) break

}
}
s; cost

Ошибка : Ошибка в s [j] <- s [j] + 1: NA не допускаются в присвоениях с подпиской </p>

Пожалуйста, помогите кому-нибудь разрешить эти ошибки.

1 Ответ

0 голосов
/ 14 мая 2018

Мы могли бы применить функцию напрямую, зациклившись на list. Обратите внимание, что каждый элемент list является matrix

sapply(lst, is.matrix)
#  1    2 
#TRUE TRUE

так, нет необходимости unlist и создавать matrix

out <- lapply(lst, nlargest, data = 40)

-проверка с результатами ОП 

out1 <- nlargest(sub1, 40)
identical(out[[1]], out1)
#[1] TRUE

Update2

На основании второго обновления нам нужно инициализировать элементы 'cost' и 'sl' с той же длиной, что и элементы 'k'. Здесь мы инициализируем sl как list из vectors 

sl <- rep(list(c(1, 1, 1, 1)), 2)
C <- 150
cost <- numeric(2)
for (k in 1:2){

for (j in mat[,,k][40:1, 1] ) {
    sl[[k]][j] <- sl[[k]][j]+1  
    cost[k] <- sum(ch*sl[[k]])
    if (cost[k] >=C) break


    }
   }   


sl
#[[1]]
#[1] 5 7 6 4

#[[2]]
#[1] 6 5 5 7

cost
#[1] 154 150
...