The matplotlib library has a module Sankey that allows to make basic Sankey Diagrams. This is an example taken from the corresponding documentation where you will find more examples of this type.
These are the parameters used in the example:
flows: Array of flow values. By convention, inputs are positive and outputs are negative.labels: List of labels for the flows (or a single label to be used for all flows).orientations: List of orientations of the flows (or a single orientation to be used for all flows). Valid values are 0 (inputs from the left, outputs to the right), 1 (from and to the top) or -1 (from and to the bottom).
# Libraries
import numpy as np
import matplotlib.pyplot as plt
from matplotlib.sankey import Sankey
# basic sankey chart
Sankey(flows=[0.25, 0.15, 0.60, -0.20, -0.15, -0.05, -0.50, -0.10], labels=['', '', '', 'First', 'Second', 'Third', 'Fourth', 'Fifth'], orientations=[-1, 1, 0, 1, 1, 1, 0,-1]).finish()
plt.title("Sankey diagram with default settings")
plt.show()
