Добавить количество узлов в диаграмму Plotly Sankey - PullRequest
Новая
       80

Добавить количество узлов в диаграмму Plotly Sankey

0 голосов
/ 30 января 2020

Я хотел бы добавить число узлов к каждому узлу на диаграмме Плотного Санки (https://plot.ly/python/sankey-diagram/), чтобы оно выглядело как число, на которое ссылаются красные стрелки. enter image description here

Возможно ли это? Я не могу найти пример этого в заговоре. Пример, который я привел выше, взят из библиотеки в R (https://github.com/fbreitwieser/sankeyD3/blob/master/README.md), но я использую график в Python.

Ниже приведен мой код. 

import plotly.graph_objects as go
import pandas as pd

def plot_sankey(df, title):
    # Get column names
    cat_columns = [key for key, value in df.dtypes.iteritems() if value == 'O']

    # Mapping to unique values for categorical columns
    labels = pd.unique(df[cat_columns].values.ravel('K')).tolist()

    # Break dowmn each step
    final = []
    for i, row in df.iterrows():
        cat_values = row[cat_columns].values
        value = row['value']
        final.extend([(a, b, value) for a, b in zip(cat_values, cat_values[1:]) if a!=None and b!=None])

    # Build formatted version
    df_formatted = pd.DataFrame(final, columns=['source', 'target', 'value'])


    # Build Node
    node = dict(
              pad = 15,
              thickness = 20,
              line = dict(color = "black", width = 0.5),
              label = labels,
              color = "blue"
            )

    # Build Link
    link = dict(
              source = [labels.index(x) for x in df_formatted['source'].tolist()],
              target = [labels.index(x) for x in df_formatted['target'].tolist()],
              value = df_formatted['value'].tolist()
          )

    # Plot
    fig = go.Figure(data=[go.Sankey(node=node, link=link, visible=True)])
    fig.update_layout(title_text=title, 
                      font_size=10,
                      autosize=False,
                      width=2750,
                      height=1600)
    fig.show()
...