python networkx网络分析库

1、载入networkx包

import networkx as nx 

2、创建网络

2.1 创建空的无向图

G = nx.Graph()

2.2 添加两类节点

#分别用不同颜色表示两类节点
for item in list_user1:
    G.add_node(item,node_color = "b")       
for item in list_user2:
    G.add_node(item,node_color  = "r")

2.3 添加边

从一个dataframe中读取数据，对应的两列应该生成对应的边，添加边。

for i in range(len(df_question_test)):
    user_id_temp = df_question_test.loc[i,"user1"]
    doctor_id_temp = df_question_test.loc[i,"user2"]
    G.add_edge(user_id_temp,doctor_id_temp)  

2.4 查看网络节点和边的数量

print(G.number_of_edges())                  
print(G.number_of_nodes())

3、网络可视化

pos=nx.spring_layout