Hierarchical tree diagram in R

Question

I am looking for an R function to build the specific type of diagram shown here. Data.tree seems promising, but I'm stuck.

My goal (shown in the image) is merely a hierarchical diagram showing counts of values in different categories from a dataframe, getting more specific as you go down. It is not a decision tree or a flow chart. What's important is something that will tabulate the count of features in each category given a dataframe and the variables I want in each level.

here's a sample of my data:

tree_data = data.frame(Context = c("urban", "rural", "urban", "urban", "rural", "rural"),
                Lighting = c("daylight", "dark", "dark", "daylight", "daylight", "dark"),
                Driver_age = c("Senior", "Adult", "Adult", "Adult", "Adult", "Senior"))

And the desired output:

I have gotten this far with data.tree:

tree_data$pathString = paste("crashes",
                             tree_data$Context,
                             tree_data$Lighting,
                             tree_data$Driver_age,
                             sep = "/")

crashes = as.Node(tree_data)
print(crashes)

The result is nicely organized, but I'm not sure how to add counts, or get it into a visual format like above.

            levelName
1  crashes           
2   ¦--urban         
3   ¦   ¦--daylight  
4   ¦   ¦   ¦--Senior
5   ¦   ¦   °--Adult 
6   ¦   °--dark      
7   ¦       °--Adult 
8   °--rural         
9       ¦--dark      
10      ¦   ¦--Adult 
11      ¦   °--Senior
12      °--daylight  
13          °--Adult

Can someone advise on next steps? Or if there is a better package, I'm open to it. I also tried diagrammeR and igraph but they did not seem like solutions that I'd be able to easily apply to different datasets. I'll need this to be something easily repeatable.

I just changed the title of this thread to reflect the fact that it is no longer about data.tree. — Kelly
– Kelly, Commented Feb 15, 2024 at 18:46

TarJae · Accepted Answer · 2024-02-13 19:14:10Z

3

We could do this easily with vtree, See the documentation https://cran.r-project.org/web/packages/vtree/vignettes/vtree.html

Here is an example:

#install.packages("vtree")

library(vtree)
tree_data = data.frame(Context = c("urban", "rural", "urban", "urban", "rural", "rural"),
                       Lighting = c("daylight", "dark", "dark", "daylight", "daylight", "dark"),
                       Driver_age = c("Senior", "Adult", "Adult", "Adult", "Adult", "Senior"))

vtree(tree_data, c("Context", "Lighting"))

answered Feb 13, 2024 at 19:14

TarJae

80.2k6 gold badges30 silver badges94 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

Kelly Over a year ago

Exactly what I was hoping for. Thank you!!

Collectives™ on Stack Overflow

Hierarchical tree diagram in R

1 Answer 1

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related