Spaces:
Runtime error
Runtime error
File size: 1,324 Bytes
0a7a9b6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
import pandas as pd
import plotly.graph_objects as go
def create_sunburst_plot(df):
fig = go.Figure(go.Sunburst(
labels=df['labels'],
parents=df['parents'],
values=df['values'],
ids=df['ids'],
text=df['text'],
hoverinfo="label+value",
branchvalues="total",
))
fig.update_layout(margin=dict(t=0, l=0, r=0, b=0))
return fig
data = [
{'ids': 'Root', 'labels': 'Root', 'parents': '', 'values': None, 'text': 'Root'},
{'ids': 'Hip Surgery', 'labels': 'Hip Surgery', 'parents': 'Root', 'values': 30, 'text': 'Hip Surgery'},
{'ids': 'Knee Surgery', 'labels': 'Knee Surgery', 'parents': 'Root', 'values': 40, 'text': 'Knee Surgery'},
{'ids': 'CPT1', 'labels': 'CPT1', 'parents': 'Hip Surgery', 'values': 20, 'text': 'CPT1'},
{'ids': 'CPT2', 'labels': 'CPT2', 'parents': 'Hip Surgery', 'values': 10, 'text': 'CPT2'},
{'ids': 'CPT3', 'labels': 'CPT3', 'parents': 'Knee Surgery', 'values': 25, 'text': 'CPT3'},
{'ids': 'CPT4', 'labels': 'CPT4', 'parents': 'Knee Surgery', 'values': 15, 'text': 'CPT4'},
]
df = pd.DataFrame(data)
# Filter DataFrame using a query parameter
def filter_data(df, query):
return df.query(query)
filtered_df = filter_data(df, "parents == 'Root'")
fig = create_sunburst_plot(filtered_df)
fig.show() |