plotly sankey graph data formatting

by
  • the structure is clearly a pandas dataframe constructor format
  • create a dataframe from it, plus the key series of the nodes
  • from this it’s simple to construct a Sankey plot from it
import pandas as pd
import numpy as np
import plotly.graph_objects as go

links = [
    {'source': 'start', 'target': 'A', 'value': 2},
    {'source': 'A', 'target': 'B', 'value': 1},
    {'source': 'A', 'target':'C', 'value':.5}

]

df = pd.DataFrame(links)
nodes = np.unique(df[["source","target"]], axis=None)
nodes = pd.Series(index=nodes, data=range(len(nodes)))

go.Figure(
    go.Sankey(
        node={"label": nodes.index},
        link={
            "source": nodes.loc[df["source"]],
            "target": nodes.loc[df["target"]],
            "value": df["value"],
        },
    )
)

enter image description here

More Related Contents:

  1. Pandas conditional creation of a series/dataframe column
  2. How to split a dataframe string column into two columns?
  3. Annotate bars with values on Pandas bar plots
  4. Comparing two dataframes and getting the differences [duplicate]
  5. Why isn’t my Pandas ‘apply’ function referencing multiple columns working? [closed]
  6. Pandas: How to easily share a sample dataframe using df.to_dict()?
  7. Accessing Pandas column using squared brackets vs using a dot (like an attribute)
  8. Get the name of a pandas DataFrame
  9. Count number of non-NaN entries in each column of Spark dataframe with Pyspark
  10. pandas concat generates nan values
  11. Pandas Groupby and Sum Only One Column
  12. What is the difference between join and merge in Pandas?
  13. Ambiguity in Pandas Dataframe / Numpy Array “axis” definition
  14. Pandas Dataframe: Replacing NaN with row average
  15. How to read a .xlsx file using the pandas Library in iPython?
  16. Pandas Replace NaN with blank/empty string
  17. How to append rows in a pandas dataframe in a for loop?
  18. ValueError: Length of values does not match length of index | Pandas DataFrame.unique()
  19. How to reorder indexed rows based on a list in Pandas data frame
  20. Trying to merge 2 dataframes but get ValueError
  21. How can I count the number of consecutive TRUEs in a DataFrame?
  22. Max and Min date in pandas groupby
  23. Pandas rolling apply using multiple columns
  24. making matplotlib scatter plots from dataframes in Python’s pandas
  25. Pandas groupby and aggregation output should include all the original columns (including the ones not aggregated on)
  26. Interactive matplotlib figures in Google Colab
  27. Replace NaN with empty list in a pandas dataframe
  28. Get weekday/day-of-week for Datetime column of DataFrame
  29. Get max value from row of a dataframe in python [duplicate]
  30. When is it appropriate to use df.value_counts() vs df.groupby(‘…’).count()?

Leave a Comment