python - 使用 CSV 文件创建思维导图

Question

我想使用带有列的 CSV 文件创建我已发送和接收的资金的思维导图：从、到、资金输入和资金输出。

我想创建一个图表，显示资金流入或流出每个账户，我想选择的账户位于中心。

我找不到执行此操作的 python 库。有图书馆可以让我这样做吗？或者甚至可以使用 Python 吗？

Answer 1

from graphviz import Digraph
import pandas as pd
import numpy as np

G = Digraph(format='jpeg')

G.attr(rankdir='LR', size='8,5')
G.attr('node', shape='circle')

df = pd.read_csv('so.csv')

# add the vertices
[G.node(str(x)) for x in np.unique(df[['From', 'To']].values.flatten())]
# add the edges
[G.edge(str(x[1][0]), str(x[1][1]), label=str(x[1][2])) for x in df.iterrows()]

G.render('sg', view=True)

其中的内容so.csv是：

From,To,Amount
Account1,Account2,20
Account1,Account3,50
Account3,Account1,60

你将会有：