[PYTHON] (Persönliches Memo) Sanky Diagramm

Zweck dieses Artikels

Erstellen Sie ein "Sankey-Diagramm" wie das folgende. {0,1} repräsentiert zwei Arten von Ereignissen, und unter der Annahme, dass mehrere Ereignisse in einem bestimmten Zeitraum auftreten, wie z. B. 0-> 1-> 1, ist der Übergang des Ereignisses Der Zweck ist zu visualisieren.

Vorbereitung

import numpy as np
import pandas as pd
pd.set_option('display.max_columns', 100)
import random

import warnings
warnings.filterwarnings('ignore')

Dummy-Datenaufbereitung

n=100

agent = [random.choice([1,2,3,4,5,6,7,8,9,10]) for i in range(n)]
day = [random.choice([1,2,3,4,5]) for i in range(n)]
time = [random.random() for i in range(n)]
event = [random.choice([0,1]) for i in range(n)]


df = pd.DataFrame({"agent":agent,
                   "day":day,
                   "time":time,
                   "event":event})

df = df.sort_values(["day","time"])

summary = df.groupby(["agent", "day"])["event"].apply(lambda x: [str(xi) for xi in x]).apply(''.join).reset_index()
summary.head()

Wie der "Agent" (Mensch oder Maschine) während eines "Tages" -Ereignisses ein Ereignis auslöste Es sind die Daten.

max_len_event = max([len(i) for i in summary["event"]])
max_len_event
>>5

summary["event"] = ["{0:-<5}".format(i) for i in summary["event"]]
summary.head()

value_list = []

for i in range(max_len_event-1):
    tmp = [ei[i:(i+2)] for ei in summary["event"]]

    #0->0
    n = sum([1 if ei=="00" else 0 for ei in tmp])
    value_list.append(n)

    #0->1
    n = sum([1 if ei=="01" else 0 for ei in tmp])
    value_list.append(n)

    #1->0
    n = sum([1 if ei=="10" else 0 for ei in tmp])
    value_list.append(n)

    #1->1
    n = sum([1 if ei=="11" else 0 for ei in tmp])
    value_list.append(n)

source_list = [[i, i] for i in range(max_len_event*2-2)]
source_list = np.array(source_list).flatten()

target_list = [[i, i+1, i, i+1] for i in range(2, max_len_event*2, 2)]
target_list = np.array(target_list).flatten()

label_list = ["0", "1"] * max_len_event

import plotly.graph_objects as go

fig = go.Figure(data=[go.Sankey(
    node = dict(
      pad = 15,
      thickness = 20,
      line = dict(color = "black", width = 0.5),
      label =  label_list,
      color = "blue"
    ),
    link = dict(
      source = source_list,
      target = target_list,
      value = value_list
  ))])

fig.update_layout(title_text="Basic Sankey Diagram", font_size=10)
fig.show()

das ist alles!

Referenz

Qiita: [Python] Was machen Sie mit der Visualisierung von 4 oder mehr Variablen? Plotly：Sankey Diagram in Python