Как эффективно преобразовать фрейм данных в список списков произвольной длины?

Question

Я пытаюсь изменить кадр данных для более эффективного хранения и поиска. Каждая строка содержит «родительское» (ключевое) значение, которое не является уникальным между строками и дочерним значением (фактически, набор из 3 атрибутов - 1 символ и 2 цифры). Я хочу преобразовать этот фрейм данных в список, который содержит только одну запись верхнего уровня для каждого уникального родительского ключа, а количество подписок, определяемое количеством дочерних элементов, ассоциируется с родителем. Вот некоторые примерные данные:

pcm <- data.frame(parent = c("middle", "middle", "might", "might", 
        "might", "million", "million", "millions"), 
       child = c("of", "school", "be", "have", "not", "in", 
        "to", "of"), 
       count = c(476, 165, 1183, 619, 321, 490, 190, 269)) 

Выход для этого должен быть список с 4-х элементов верхнего уровня (называемых "middle", "might", "million", "millions") и различным числом подсписков с именованными членами $child и $count (например lookup4[["middle"]] содержит суб -листов $children[[1]]$child = "of", $count = 476 и $children[[2]]$child = "school", $count = 165).

Приведенный ниже код работает, но очень медленный (несколько часов в кадре данных 300 000 строк с использованием 8 ГБ ОЗУ). Я наложил ограничение 6 на количество детей в выходных данных, но, похоже, это не имело большого значения.

lookup4 <- list() 
parents <- unique(pcm$parent) 
n.parents <- length(parents) 
for (i in 1:n.parents) { 
    words <- pcm$child[pcm$parent == parents[i]] 
    counts <- pcm$count[pcm$parent == parents[i]] 
    probs <- pcm$prob[pcm$parent == parents[i]] 
    n.children <- min(c(NROW(words), 6) 
    ngram.tail <- list() 
    for (k in 1:n.children) { 
     ngram.tail[[k]] <- list(word = words[k], 
     count = counts[k], 
     prob = probs[k]) 
    } 
    lookup4[[parents[i]]] <- list(children = ngram.tail) 
} 

Могу ли я ускорить его, исключив цикл «для»? Если да, то как мне закодировать трансформацию?

Answer 1

Попробуйте это:

Я полагаю, что dataframe называется parents:

parents.list <- as.list(as.data.frame(t(parents))) 

Если вы хотите, имена строк родителей, чтобы быть имена в списке:

parents.list <- setNames(split(parents, seq(nrow(parents))), rownames(parents))