Update sankey-diagram.md

Author: LiamConnors

Diff for: doc/python/sankey-diagram.md

@@ -218,42 +218,22 @@ Create a Sankey diagram with arrow links by setting the `arrowlen` attribute of
 
 ```python
 import plotly.graph_objects as go
-import urllib, json
-
-url = 'https://raw.githubusercontent.com/plotly/plotly.js/master/test/image/mocks/sankey_energy.json'
-response = urllib.request.urlopen(url)
-data = json.loads(response.read())
-
-# override gray link colors with 'source' colors
-opacity = 0.4
-# change 'magenta' to its 'rgba' value to add opacity
-data['data'][0]['node']['color'] = ['rgba(255,0,255, 0.8)' if color == "magenta" else color for color in data['data'][0]['node']['color']]
-data['data'][0]['link']['color'] = [data['data'][0]['node']['color'][src].replace("0.8", str(opacity))
-                                    for src in data['data'][0]['link']['source']]
 
-fig = go.Figure(data=[go.Sankey(
-    valueformat = ".0f",
-    valuesuffix = "TWh",
-    # Define nodes
+fig = go.Figure(go.Sankey(
+    arrangement = "snap",
     node = dict(
-      pad = 15,
-      thickness = 15,
-      line = dict(color = "black", width = 0.5),
-      label =  data['data'][0]['node']['label'],
-      color =  data['data'][0]['node']['color']
+        label = ["A", "B", "C", "D", "E", "F"],
+        x = [0.2, 0.1, 0.5, 0.7, 0.3, 0.5],
+        y = [0.7, 0.5, 0.2, 0.4, 0.2, 0.3],
+        pad = 10  
     ),
-    # Add links
     link = dict(
-      arrowlen = 15,
-      source =  data['data'][0]['link']['source'],
-      target =  data['data'][0]['link']['target'],
-      value =  data['data'][0]['link']['value'],
-      label =  data['data'][0]['link']['label'],
-      color =  data['data'][0]['link']['color']
-))])
+        arrowlen = 15,
+        source = [0, 0, 1, 2, 5, 4, 3, 5],
+        target = [5, 3, 4, 3, 0, 2, 2, 3],
+        value = [1, 2, 1, 1, 1, 1, 1, 2]  
+    )))
 
-fig.update_layout(title_text="Energy forecast for 2050<br>Source: Department of Energy & Climate Change, Tom Counsell via <a href='https://bost.ocks.org/mike/sankey/'>Mike Bostock</a>",
-                  font_size=10)
 fig.show()
 ```