Plotly:如何在 Sankey 图中设置节点位置? [英] Plotly: How to set node positions in a Sankey Diagram?

问题描述

样本数据如下:

unique_list = ['home0', 'page_a0', 'page_b0', 'page_a1', 'page_b1','page_c1', 'page_b2', 'page_a2', 'page_c2', 'page_c3']来源 = [0, 0, 1, 2, 2, 3, 3, 4, 4, 7, 6]目标 = [3, 4, 4, 3, 5, 6, 8, 7, 8, 9, 9]值 = [2, 1, 1, 1, 1, 2, 1, 1, 1, 1, 2]

使用来自

但是，我想获得在同一垂直列中以相同数字结尾的所有值，就像最左边的列的所有节点都以 0 结尾一样.我明白了在

请注意，在此示例中未明确设置 y 值.一旦一个共同的 x 值有多个节点，y 值将自动调整，以便所有节点显示在同一垂直位置.如果您确实想明确设置所有位置，只需设置arrangement='fixed'

我添加了一个自定义函数 nodify()，它为具有共同结尾的标签名称分配相同的 x 位置，例如 中的 '0'['home0', 'page_a0', 'page_b0'].现在，如果您将 page_c1 更改为 page_c2 作为示例，您将得到:

完整代码:

import plotly.graph_objects as gounique_list = ['home0', 'page_a0', 'page_b0', 'page_a1', 'page_b1','page_c1', 'page_b2', 'page_a2', 'page_c2', 'page_c3']来源 = [0, 0, 1, 2, 2, 3, 3, 4, 4, 7, 6]目标 = [3, 4, 4, 3, 5, 6, 8, 7, 8, 9, 9]值 = [2, 1, 1, 1, 1, 2, 1, 1, 1, 1, 2]定义节点(节点名称):node_names = unique_list# 唯一名称结尾结束 = 排序(列表(设置([e[-1] for e in node_names])))# 间隔步数 = 1/len(结束)# 每个唯一名称结尾的 x 值# 作为节点位置的输入节点_x = {}xVal = 0对于 e 结束:nodes_x[str(e)] = xValxVal += 步骤# 列表形式的 x 和 y 值x_values = [nodes_x[n[-1]] for n in node_names]y_values = [0.1]*len(x_values)返回 x_values, y_valuesnodified = nodify(node_names=unique_list)# 情节设置fig = go.Figure(data=[go.Sankey(安排='快照'，节点 = 字典(垫 = 15,厚度 = 20,line = dict(color = "black", width = 0.5),标签 = unique_list，颜色=蓝色"，x=nodified[0],y=已确认[1]),链接 = 字典(来源 = 来源，目标 = 目标，价值 = 价值))])图.show()

The sample data is as follows:

unique_list = ['home0', 'page_a0', 'page_b0', 'page_a1', 'page_b1', 
               'page_c1', 'page_b2', 'page_a2', 'page_c2', 'page_c3']
sources = [0, 0, 1, 2, 2, 3, 3, 4, 4, 7, 6]
targets = [3, 4, 4, 3, 5, 6, 8, 7, 8, 9, 9]
values = [2, 1, 1, 1, 1, 2, 1, 1, 1, 1, 2]

Using the sample code from the documentation

fig = go.Figure(data=[go.Sankey(
    node = dict(
      pad = 15,
      thickness = 20,
      line = dict(color = "black", width = 0.5),
      label = unique_list,
      color = "blue"
    ),
    link = dict(
      source = sources,
      target = targets,
      value = values
  ))])

fig.show()

This outputs the following sankey diagram

However, I would like to get all the values which end in the same number in the same vertical column, just like how the leftmost column has all of it's nodes ending with a 0. I see in the docs that it is possible to move the node positions, however I was wondering if there was a cleaner way to do it other than manually inputting x and y values. Any help appreciated.

解决方案

In go.Sankey() set arrangement='snap' and adjust x and y positions in x=<list> and y=<list>. The following setup will place your nodes as requested.

Plot:

Please note that the y-values are not explicitly set in this example. As soon as there are more than one node for a common x-value, the y-values will be adjusted automatically for all nodes to be displayed in the same vertical position. If you do want to set all positions explicitly, just set arrangement='fixed'

Edit:

I've added a custom functin nodify() that assigns identical x-positions to label names that have a common ending such as '0' in ['home0', 'page_a0', 'page_b0']. Now, if you as an example change page_c1 to page_c2 you'll get this:

Complete code:

import plotly.graph_objects as go
unique_list = ['home0', 'page_a0', 'page_b0', 'page_a1', 'page_b1', 
               'page_c1', 'page_b2', 'page_a2', 'page_c2', 'page_c3']
sources = [0, 0, 1, 2, 2, 3, 3, 4, 4, 7, 6]
targets = [3, 4, 4, 3, 5, 6, 8, 7, 8, 9, 9]
values = [2, 1, 1, 1, 1, 2, 1, 1, 1, 1, 2]


def nodify(node_names):
    node_names = unique_list
    # uniqe name endings
    ends = sorted(list(set([e[-1] for e in node_names])))
    
    # intervals
    steps = 1/len(ends)

    # x-values for each unique name ending
    # for input as node position
    nodes_x = {}
    xVal = 0
    for e in ends:
        nodes_x[str(e)] = xVal
        xVal += steps

    # x and y values in list form
    x_values = [nodes_x[n[-1]] for n in node_names]
    y_values = [0.1]*len(x_values)
    
    return x_values, y_values

nodified = nodify(node_names=unique_list)

# plotly setup
fig = go.Figure(data=[go.Sankey(
      arrangement='snap',
      node = dict(
      pad = 15,
      thickness = 20,
      line = dict(color = "black", width = 0.5),
      label = unique_list,
      color = "blue",
     x=nodified[0],
     y=nodified[1]
    ),
    link = dict(
      source = sources,
      target = targets,
      value = values
  ))])

fig.show()

这篇关于Plotly:如何在 Sankey 图中设置节点位置?的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！