import pandas as pd
import holoviews as hv

hv.extension('matplotlib')
hv.output(fig='svg')

edges = pd.read_csv('../../../assets/energy.csv')
edges.head(5)

hv.Sankey(edges, label='Energy Diagram').opts(
    label_position='left', edge_color='target', node_color='index', cmap='tab20')