Jak przekonwertować ramkę danych na zagnieżdżoną listę?

Question

Mam ramkę danych, którą chcę przekonwertować na listę zagnieżdżoną z niestandardowym poziomem zagnieżdżenia. Oto jak to zrobić, ale jestem pewien, że istnieje lepszy sposób:

data <- data.frame(city=c("A", "A", "B", "B"), street=c("a", "b", "a", "b"), tenant=c("Smith","Jones","Smith","Jones"), income=c(100,200,300,400)) 

nested_data <- lapply(levels(data$city), function(city){ 
    data_city <- subset(data[data$city == city, ], select=-city) 
    list(city = city, street_values=lapply(levels(data_city$street), function(street){ 
     data_city_street <- subset(data_city[data_city$street == street, ], select=-street) 
     tenant_values <- apply(data_city_street, 1, function(income_tenant){ 
      income_tenant <- as.list(income_tenant) 
      list(tenant=income_tenant$tenant, income=income_tenant$income) 
     }) 
     names(tenant_values) <- NULL 
     list(street=street, tenant_values=tenant_values) 
    })) 
}) 

Wyjście w JSON wygląda następująco:

library(rjson) 
write(toJSON(nested_data), "") 
[{"city":"A","street_values":[{"street":"a","tenant_values":[{"tenant":"Smith","income":"100"}]},{"street":"b","tenant_values":[{"tenant":"Jones","income":"200"}]}]},{"city":"B","street_values":[{"street":"a","tenant_values":[{"tenant":"Smith","income":"300"}]},{"street":"b","tenant_values":[{"tenant":"Jones","income":"400"}]}]}] 

# or prettified: 

[ 
    { 
    "city": "A", 
    "street_values": [ 
     { 
     "street": "a", 
     "tenant_values": [ 
      { 
      "tenant": "Smith", 
      "income": "100" 
      } 
     ] 
     }, 
     { 
     "street": "b", 
     "tenant_values": [ 
      { 
      "tenant": "Jones", 
      "income": "200" 
      } 
     ] 
     } 
    ] 
    }, 
    { 
    "city": "B", 
    "street_values": [ 
     { 
     "street": "a", 
     "tenant_values": [ 
      { 
      "tenant": "Smith", 
      "income": "300" 
      } 
     ] 
     }, 
     { 
     "street": "b", 
     "tenant_values": [ 
      { 
      "tenant": "Jones", 
      "income": "400" 
      } 
     ] 
     } 
    ] 
    } 
] 

Czy istnieje lepszy sposób to zrobić?

Answer 1

Co na temat korzystania split, aby dostać się większość drogi i rapply do ostatniego kroku:

nestedList <- rapply(lapply(split(data[-1], data[1]), 
          function(x) split(x[-1], x[1])), 
        f = function(x) as.character(unlist(x)), 
        how = "replace") 

Oto wynik:

nestedList 
# $A 
# $A$a 
# $A$a$tenant 
# [1] "Smith" 
# 
# $A$a$income 
# [1] "100" 
# 
# 
# $A$b 
# $A$b$tenant 
# [1] "Jones" 
# 
# $A$b$income 
# [1] "200" 
# 
# 
# 
# $B 
# $B$a 
# $B$a$tenant 
# [1] "Smith" 
# 
# $B$a$income 
# [1] "300" 
# 
# 
# $B$b 
# $B$b$tenant 
# [1] "Jones" 
# 
# $B$b$income 
# [1] "400" 

a struktura:

> str(nestedList) 
List of 2 
$ A:List of 2 
    ..$ a:List of 2 
    .. ..$ tenant: chr "Smith" 
    .. ..$ income: chr "100" 
    ..$ b:List of 2 
    .. ..$ tenant: chr "Jones" 
    .. ..$ income: chr "200" 
$ B:List of 2 
    ..$ a:List of 2 
    .. ..$ tenant: chr "Smith" 
    .. ..$ income: chr "300" 
    ..$ b:List of 2 
    .. ..$ tenant: chr "Jones" 
    .. ..$ income: chr "400" 

Struktura nie jest dokładnie taka, jakiej szukasz, ale może to pomóc zacząć od alternatywnego podejścia.

Answer 2

znalazłem rozwiązanie na moje pytanie o zmiany funkcji jak:

nestedList <- rapply(lapply(split(df[-1], df[1]), 
          function(x) split(x[-1], x[1])), 
        f = function(x) as.data.frame(as.list(split(x,x))), how = "replace")