sunburst.py

import plotly.express as px
import pandas as pd

def generate_df():
    data = {
        "names": ["alice", "bob", "chuck", "david", "ella", "fred"],
        "parents": ["", "alice", "bob", "chuck", "david", "david"],
    }
    return pd.DataFrame(data)


def plot():
    px.sunburst(
        generate_df(),
        names=["alice", "bob", "chuck", "david", "ella", "fred"],
        parents=["", "alice", "bob", "chuck", "david", "david"],
        values=[0, 0, 1, 0, 1, 1]
        ).show()