社交图：Pandas 数据框到 Networkx 图

Question

我有一个数据集，其中包含一个用户和他们拥有的每个朋友。格式与下面类似，其中有一个user_id 字段和friend_id 字段，其中包含每个朋友的相应ID。

user_id | friend_id
   A          B
   A          C
   B          A
   B          G

我的目标是得到一个无向图，显示每个用户和他们喜欢的每个朋友之间的边缘。

A - B - G
|
C

我很难找到如何将 pandas 链接到 networkx 或 graphviz 以及其他扩展从表格数据创建社交图的资源。

Answer 1

例如，这里是一种使用networkx、pandas 和matplotlib 显示无向网络图的方法。

代码：

import matplotlib.pyplot as plt
import networkx as nx

# store pairs to a list of tuples
tuples = [tuple(x) for x in df.values]

# set up a graph and show it
G = nx.DiGraph()
G.add_edges_from(tuples)
nx.draw_networkx(G)
plt.xticks([], [])
plt.yticks([], [])
plt.show()

输出：