Colonna di lista degli zoccoli nel frame di dati con colonna ID

Question

Il mio riquadro dati contiene l'output di un sondaggio con un tipo di domanda multipla selezionato. Alcune celle hanno più valori.

df <- data.frame(a=1:3,b=I(list(1,1:2,1:3))) 
df 
    a  b 
1 1  1 
2 2 1, 2 
3 3 1, 2, 3 

vorrei appiattire la lista per ottenere il seguente output:

df 
    a  b 
1 1  1 
2 2  1 
3 2  2 
4 3  1 
5 3  2 
6 3  3 

dovrebbe essere facile, ma in qualche modo non riesco a trovare i termini di ricerca. Grazie.

Answer 1

si può semplicemente utilizzare unnest da "tidyr":

library(tidyr) 
unnest(df, b) 
# a b 
# 1 1 1 
# 2 2 1 
# 3 2 2 
# 4 3 1 
# 5 3 2 
# 6 3 3 

Answer 2

Utilizzando base R, una possibilità è stack dopo aver nominato i list elementi della colonna 'b' con quella degli elementi di 'a'. Possiamo usare setNames per cambiare i nomi.

stack(setNames(df$b, df$a)) 

o un'altra opzione sarebbe quella di utilizzare unstack di nominare automaticamente l'elemento elenco di 'b' con elementi 'a' e poi fare il stack per ottenere un'uscita data.frame.

stack(unstack(df, b~a)) 

---

Oppure si può usare una comoda funzione listCol_l da splitstackshape per convertire il list-data.frame.

library(splitstackshape) 
listCol_l(df, 'b') 

Answer 3

Ecco un modo, con data.table:

require(data.table) 
data.table(df)[,as.integer(unlist(b)),by=a] 

Se b viene memorizzato in modo coerente, as.integer può essere saltata. È possibile verificare con

unique(sapply(df$b,class)) 
# [1] "numeric" "integer" 

Answer 4

Ecco un'altra soluzione di base, molto meno elegante di qualsiasi altra soluzione pubblicata finora. Pubblicare per completezza, anche se personalmente consiglierei la soluzione di base di akrun.

with(df, cbind(a = rep(a, sapply(b, length)), b = do.call(c, b))) 

Questo costruisce la prima colonna come gli elementi di a, dove ognuno viene ripetuto in base alla lunghezza dell'elemento di elenco corrispondente dal b. La seconda colonna è b "appiattita" utilizzando do.call() con c().

Come indicato da Ananda Mahto in un commento, sapply(b, length) può essere sostituito con lengths(b) nella versione più recente di R (3.2, se non sbaglio).