Python 绘制惊艳的桑基图，你学会了吗？（2）

　　NODES = dict( 

　　#         0                           1                             2        3       4         5                         

　　label = ["United States of America", "People's Republic of China", "Japan", "Gold", "Silver", "Bronze"],

　　color = ["seagreen",                 "dodgerblue",                 "orange", "gold", "silver", "brown" ],)

　　x = [     0,                          0,                            0,        0.5,    0.5,      0.5],

　　y = [     0,                          0.5,                          1,        0.1,    0.5,        1],)

　　data = go.Sankey(node = NODES, link = LINKS)

　　fig = go.Figure(data)

　　fig.update_layout(title="Olympics - 2021: Country &  Medals",  font_size=16)

　　fig.show()

　　NODES = dict( 

　　#         0                           1                               2        3       4           5

　　label = ["United States of America", "People's Republic of China",   "Japan", "Gold", "Silver", "Bronze"],

　　color = [                "seagreen",                 "dodgerblue",  "orange", "gold", "silver", "brown" ],

　　x     = [                         0,                            0,         0,    0.5,      0.5,      0.5],

　　y     = [                         0,                          0.5,         1,    0.1,      0.5,        1],

　　hovertemplate=" ",)

　　LINK_LABELS = []

　　for country in ["USA","China","Japan"]:

　　    for medal in ["Gold","Silver","Bronze"]:

　　        LINK_LABELS.append(f"{country}-{medal}")

　　LINKS = dict(source = [  0,  0,  0,  1,  1,  1,  2,  2,  2], 

　　       # 链接的起点或源节点

　　       target = [  3,  4,  5,  3,  4,  5,  3,  4,  5], 

　　       # 链接的目的地或目标节点

　　       value =  [ 39, 41, 33, 38, 32, 18, 27, 14, 17], 

　　       # 链接的宽度（数量） 

　　             # 链接的颜色

　　             # 目标节点：3-Gold          4 -Silver        5-Bronze

　　             color = ["lightgreen",   "lightgreen",   "lightgreen",   # 源节点：0 - 美国

　　                      "lightskyblue", "lightskyblue", "lightskyblue", # 源节点：1 - 中国

　　                      "bisque",       "bisque",       "bisque"],      # 源节点：2 - 日本

　　             label = LINK_LABELS, 

　　             hovertemplate="%{label}",)

　　data = go.Sankey(node = NODES, link = LINKS)

　　fig = go.Figure(data)

　　fig.update_layout(title="Olympics - 2021: Country &  Medals",  

　　                  font_size=16, width=1200, height=500,)

　　fig.update_traces(valueformat='3d', 

　　                  valuesuffix='Medals', 

　　                  selector=dict(type='sankey'))

　　fig.update_layout(hoverlabel=dict(bgcolor="lightgray",

　　                                  font_size=16,

　　                                  font_family="Rockwell"))

　　fig.show("png") #fig.show()

　　NUM_COUNTRIES = 5

　　X_POS, Y_POS = 0.5, 1/(NUM_COUNTRIES-1)

　　NODE_COLORS = ["seagreen", "dodgerblue", "orange", "palevioletred", "darkcyan"]

　　LINK_COLORS = ["lightgreen", "lightskyblue", "bisque", "pink", "lightcyan"]

　　source = []

　　node_x_pos, node_y_pos = [], []

　　node_labels, node_colors = [], NODE_COLORS[0:NUM_COUNTRIES]

　　link_labels, link_colors, link_values = [], [], [] 

　　# 第一组链接和节点

　　for i in range(NUM_COUNTRIES):

　　    source.extend([i]*3)

　　    node_x_pos.append(0.01)

　　    node_y_pos.append(round(i*Y_POS+0.01,2))

　　    country = df_medals['Country'][i]

　　    node_labels.append(country) 

　　    for medal in ["Gold", "Silver", "Bronze"]:

　　        link_labels.append(f"{country}-{medal}")

　　        link_values.append(df_medals[f"{medal} Medals"][i])

　　    link_colors.extend([LINK_COLORS[i]]*3)

　　source_last = max(source)+1

　　target = [ source_last, source_last+1, source_last+2] * NUM_COUNTRIES

　　target_last = max(target)+1

　　node_labels.extend(["Gold", "Silver", "Bronze"])

　　node_colors.extend(["gold", "silver", "brown"])

　　node_x_pos.extend([X_POS, X_POS, X_POS])

　　node_y_pos.extend([0.01, 0.5, 1])

　　# 最后一组链接和节点

　　source.extend([ source_last, source_last+1, source_last+2])

　　target.extend([target_last]*3)

　　node_labels.extend(["Total Medals"])

　　node_colors.extend(["grey"])

　　node_x_pos.extend([X_POS+0.25])

　　node_y_pos.extend([0.5])

　　for medal in ["Gold","Silver","Bronze"]:

　　    link_labels.append(f"{medal}")

　　    link_values.append(df_medals[f"{medal} Medals"][:i+1].sum())

　　link_colors.extend(["gold", "silver", "brown"])

　　print("node_labels", node_labels)

　　print("node_x_pos", node_x_pos); print("node_y_pos", node_y_pos)

　　node_labels ['United States of America', "People's Republic of China", 

　　             'Japan', 'Great Britain', 'ROC', 'Gold', 'Silver', 

　　             'Bronze', 'Total Medals']

　　node_x_pos [0.01, 0.01, 0.01, 0.01, 0.01, 0.5, 0.5, 0.5, 0.75]

　　node_y_pos [0.01, 0.26, 0.51, 0.76, 1.01, 0.01, 0.5, 1, 0.5]

　　# 显示的图

　　NODES = dict(pad  = 20, thickness = 20, 

　　             line = dict(color = "lightslategrey",

　　                         width = 0.5),

　　             hovertemplate=" ",

　　             label = node_labels, 

　　             color = node_colors,

　　             x = node_x_pos, 

　　             y = node_y_pos, )

　　LINKS = dict(source = source, 

　　             target = target, 

　　             value = link_values, 

　　             label = link_labels, 

　　             color = link_colors,

　　             hovertemplate="%{label}",)

　　data = go.Sankey(arrangement='snap', 

　　                 node = NODES, 

　　                 link = LINKS)

　　fig = go.Figure(data)

　　fig.update_traces(valueformat='3d', 

　　                  valuesuffix=' Medals', 

　　                  selector=dict(type='sankey'))

　　fig.update_layout(title="Olympics - 2021: Country &  Medals",  

　　                  font_size=16,  

　　                  width=1200,

　　                  height=500,)

　　fig.update_layout(hoverlabel=dict(bgcolor="grey", 

　　                                  font_size=14, 

　　                                  font_family="Rockwell"))

　　fig.show("png")