Plotly：如何生成并排的 px.sunburst 图？

Question

I am using the following Python code to produce a sunburst plot:-

import plotly
import plotly.express as px
import plotly.graph_objects as go
import pandas as pd
df = pd.read_csv('CellUsageStats.csv',dtype={'Population':int},keep_default_na=False)
print(df.head(50))
fig = px.sunburst(df, path=['LibPrefix', 'MasterPrefixAbrHead', 'MasterPrefixAbrTail', 'Drive'], values='Population', color='Population', color_continuous_scale=px.colors.sequential.Inferno)
fig.update_traces(hovertemplate='Pattern: %{currentPath}%{label}<br>Matches: %{value}<br>Percent of <b>%{parent}</b>: %{percentParent:0.2%f}<br>Percent of <b>%{entry}</b>: %{percentEntry:0.2%f}<br>Percent of <b>%{root}</b>: %{percentRoot:0.2%f}')
fig.show()
plotly.offline.plot(fig, filename='sunburst.html')

I have a second csv file that I use to produce another sunburst, using the same code above. How do I combine them to create side-by-side plots in the same html output? I found...

https://plotly.com/python/sunburst-charts/

However, I really need px.sunburst attributes, and the examples in the above link use go.Sunburst, not px.sunburst. Is there a straightforward way to produce side-by-side plots from two different sets of data?

Thanks!

Answer 1

Here's a modified version of @vestland's answer that makes use of subplots:

import plotly.express as px
from plotly.subplots import make_subplots

# data for px.sunburst1
data1 = dict(character=["Eve", "Cain", "Seth", "Enos", "Noam", "Abel", "Awan", "Enoch", "Azura"],
             parent=["", "Eve", "Eve", "Seth", "Seth", "Eve", "Eve", "Awan", "Eve"],
             value=[10, 14, 12, 10, 2, 6, 6, 4, 4])

# data for px.sunburst2
data2 = dict(character=["Eve", "Cain", "Seth", "Enos"],
             parent=["", "Eve", "Eve", "Seth"],
             value=[10, 14, 12, 10])

# extract data and structure FROM px.sunburst1
sb1 = px.sunburst(data1,
                  names='character',
                  parents='parent',
                  values='value',
                  )

# extract data and structure FROM px.sunburst2
sb2 = px.sunburst(data2,
                  names='character',
                  parents='parent',
                  values='value',
                  )

fig = make_subplots(rows=1, cols=2, specs=[
    [{"type": "sunburst"}, {"type": "sunburst"}]
])

fig.add_trace(sb1.data[0], row=1, col=1)
fig.add_trace(sb2.data[0], row=1, col=2)
fig.show()

Answer 2

Answer

The following suggestion will show you how you can:

1. build two separate px.Sunburst figures, and

2. extract all necessary info regarding data and structure from them using ._data , and

3. use that to build two go.Sunburst traces, and

4. set them up side by side using the domain attribute.

In other words, you form your figures using px.Sunburst , and steal what you need from there to easily set up two sunburst diagrams side by side.

Plot

Complete code

import plotly
import plotly.offline as py
import plotly.graph_objs as go
import plotly.express as px

# data for px.sunburst1
data1 = dict(character=["Eve", "Cain", "Seth", "Enos", "Noam", "Abel", "Awan", "Enoch", "Azura"],
             parent=["", "Eve", "Eve", "Seth", "Seth", "Eve", "Eve", "Awan", "Eve" ],
             value=[10, 14, 12, 10, 2, 6, 6, 4, 4])

# data for px.sunburst2
data2 = dict(character=["Eve", "Cain", "Seth", "Enos"],
             parent=["", "Eve", "Eve", "Seth" ],
             value=[10, 14, 12, 10])

# extract data and structure FROM px.sunburst1
sb1 =px.sunburst(data1,
                 names='character',
                 parents='parent',
                 values='value',
                )._data

# extract data and structure FROM px.sunburst2
sb2 =px.sunburst(data2,
                 names='character',
                 parents='parent',
                 values='value',
                )._data

# traces with separate domains to form a subplot
trace1 = go.Sunburst(labels=sb1[0]['labels'],
                        parents=sb1[0]['parents'],
                        values=sb1[0]['values'],
                        domain={'x': [0.0, 0.5], 'y': [0.0, 1]})

trace2 = go.Sunburst(labels=sb2[0]['labels'],
                        parents=sb2[0]['parents'],
                        values=sb2[0]['values'],
                        domain={'x': [0.5, 1.0], 'y': [0., 1.00]})

# layout and figure production
layout = go.Layout(height = 600,
                   width = 600,
                   autosize = False,
                   title = 'Side by side px.Sunburst diagrams')
fig = go.Figure(data = [trace1, trace2], layout = layout)
fig.show()