How to make a sunburst plot in R or Python?

by

Python version of sunburst diagram using matplotlib bars in polar projection:

import numpy as np
import matplotlib.pyplot as plt

def sunburst(nodes, total=np.pi * 2, offset=0, level=0, ax=None):
    ax = ax or plt.subplot(111, projection='polar')

    if level == 0 and len(nodes) == 1:
        label, value, subnodes = nodes[0]
        ax.bar([0], [0.5], [np.pi * 2])
        ax.text(0, 0, label, ha="center", va="center")
        sunburst(subnodes, total=value, level=level + 1, ax=ax)
    elif nodes:
        d = np.pi * 2 / total
        labels = []
        widths = []
        local_offset = offset
        for label, value, subnodes in nodes:
            labels.append(label)
            widths.append(value * d)
            sunburst(subnodes, total=total, offset=local_offset,
                     level=level + 1, ax=ax)
            local_offset += value
        values = np.cumsum([offset * d] + widths[:-1])
        heights = [1] * len(nodes)
        bottoms = np.zeros(len(nodes)) + level - 0.5
        rects = ax.bar(values, heights, widths, bottoms, linewidth=1,
                       edgecolor="white", align='edge')
        for rect, label in zip(rects, labels):
            x = rect.get_x() + rect.get_width() / 2
            y = rect.get_y() + rect.get_height() / 2
            rotation = (90 + (360 - np.degrees(x) % 180)) % 360
            ax.text(x, y, label, rotation=rotation, ha="center", va="center") 

    if level == 0:
        ax.set_theta_direction(-1)
        ax.set_theta_zero_location('N')
        ax.set_axis_off()

Example, how this function can be used:

data = [
    ("https://stackoverflow.com/", 100, [
        ('home', 70, [
            ('Images', 40, []),
            ('Videos', 20, []),
            ('Documents', 5, []),
        ]),
        ('usr', 15, [
            ('src', 6, [
                ('linux-headers', 4, []),
                ('virtualbox', 1, []),

            ]),
            ('lib', 4, []),
            ('share', 2, []),
            ('bin', 1, []),
            ('local', 1, []),
            ('include', 1, []),
        ]),
    ]),
]

sunburst(data)

python matplotlib sunburst diagram

More Related Contents:

  1. Adding value labels on a matplotlib bar chart
  2. Store large data or a service connection per Flask session
  3. How to make IPython notebook matplotlib plot inline
  4. Construct a manual legend for a complicated plot
  5. How do I convert a numpy array to (and display) an image?
  6. Heatmap in matplotlib with pcolor?
  7. seaborn is not plotting within defined subplots
  8. Moving x-axis to the top of a plot in matplotlib
  9. How to create a density plot in matplotlib?
  10. groupby weighted average and sum in pandas dataframe
  11. Use index in pandas to plot data
  12. Animated sorted bar chart with bars overtaking each other
  13. Python’s equivalent for R’s dput() function
  14. Seaborn: countplot() with frequencies
  15. What are Python pandas equivalents for R functions like str(), summary(), and head()?
  16. Plotting the data with scrollable x (time/horizontal) axis on Linux
  17. Converting python objects for rpy2
  18. R and Python in one Jupyter notebook
  19. R expand.grid() function in Python
  20. ggplot2 multiple scales/legends per aesthetic, revisited [duplicate]
  21. How to remove or hide y-axis ticklabels from a matplotlib / seaborn plot
  22. Add legend to geom_line() graph in r
  23. OSError: cannot load library ‘C:\Program Files\R\R-4.0.2\bin\x64\R.dll’: error 0x7e
  24. Circular shift of vector (equivalent to numpy.roll)
  25. Placing Custom Images in a Plot Window–as custom data markers or to annotate those markers
  26. How do Rpy2, pyrserve and PypeR compare?
  27. Align multiple ggplot2 plots with grid
  28. Combine two Pyplot patches for legend
  29. Excel, How to split cells by comma delimiter into new cells
  30. How to add hatches to boxplots with sns.boxplot or sns.catplot

Leave a Comment