import plotly
import pandas as pd
import numpy as np
import urllib, json
import plotly.graph_objs as go

plotly.offline.init_notebook_mode(connected=True)

trees_df = pd.read_csv('trees.csv', encoding='cp1252')

trees_df.head()

data_trace = dict(type='sankey', domain=dict(x=[0,1],y=[0,1]),
                 orientation="h", valueformat=".0f",
                 node = dict(pad=10, thickness=50,
                 line = dict(color="black",width=0.5),
                         label=trees_df['node_label'].dropna(axis=0, how='any'),
                         color=trees_df['node_color']),
                 link = dict(source=trees_df['source'].dropna(axis=0, how='any'),
                          target=trees_df['target'].dropna(axis=0, how='any'),
                          value=trees_df['value'].dropna(axis=0, how='any'),
                          color=trees_df['trace_color'].dropna(axis=0, how='any')))

layout = dict(title="Native Tree Species of Canada",
              font=dict(size=11))

fig = dict(data=[data_trace], layout=layout)
plotly.offline.iplot(fig, validate=False)