Spaces:
Runtime error
Runtime error
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() |