Преобразовать в цикл для применения - PullRequest
Новая
       12

Преобразовать в цикл для применения

3 голосов
/ 15 ноября 2011

В R как заменить следующий код с помощью таких функций, как apply, lapply, rapply, do.call и т. Д.? 

u <- 10:12
slist <- list()

for (i in 1:length(u)) {
  p <- combn(u, i) 
  for (j in 1:ncol(p)) {
    s <- paste(p[,j], collapse=",")
    slist[[s]] <- 0
  }
}


Для этой части: 

  for (j in 1:ncol(p)) {
    s <- paste(p[,j], collapse=",")

Я пробовал что-то вроде: 

  s <- apply(p, 2, function(x) paste(x, collapse=","))

Который работает. Но затем для этой slist[[s]] <- 0 части внутри того же цикла for, я не знаю, что делать.

Редактировать: Это то, что я пытаюсь сделать. Для вектора u я создаю список всех подмножеств в этом векторе. Затем для каждого подмножества я присваиваю его s, затем использую строку s в качестве имени элемента в slist. Странно, я знаю, но это домашнее задание. Для приведенного выше кода это будут первые 5 элементов slist: 

 > slist
 $`10`
 [1] 0

 $`11`
 [1] 0

 $`12`
 [1] 0

 $`10,11`
 [1] 0

 $`10,12`
 [1] 0

Да, я просто пытаюсь научиться правильно применять и делать вещи.

Ответы [ 3 ]

4 голосов
/ 16 ноября 2011

Вот одно из решений: 

n <- unlist(lapply(seq_along(u), function(i) {
  apply(combn(length(u),i),2, function(x) paste(u[x], collapse=','))
}
))

slist <- list()
slist[n] <- 0

ОБНОВЛЕНИЕ Написано одновременно с @djhurio, оно очень похоже, но я позволил себе сменить использование combn, чтобы оно обрабатывало u длины 1 как @djhurio указал.

3 голосов
/ 16 ноября 2011

Решение с использованием одного apply и одного lapply. Работает также, если length(u)==1. 

# Define function to create combinations
f1 <- function(y, i) {
  if (length(y)==1) as.character(y) else {
    p <- combn(y, i)
    apply(p, 2, function(x) paste(x, collapse=","))
  }
}

# Initial vector
u <- 10:12

# Character vector with all posible combinations
l <- unlist(lapply(1:length(u), f1, y=u))
l

# Create list with 0 values and same length as l
slist <- as.list(rep(0, length(l)))

# Assign names to the list objects
names(slist) <- l

slist
1 голос
/ 16 ноября 2011

Другое решение без необходимости анонимных функций. mapply векторизует combn, а rapply рекурсивно перебирает список комбинаций, сворачивая их с помощью ,. 

rapply(mapply(combn, list(u), seq_along(u), simplify = F), paste, collapse = ",")
...