我想创建一个图,如果一个特定属性相同,它会自动在节点之间添加边。图中的节点代表学生。我正在向节点添加两个属性:university_id
和{
我一直在研究这个解决方案:NetworkX: add edges to graph from node attributes
通过测试,这个解决方案似乎连接了一个图的所有边,不管这些属性是否相同。有没有一个简单的解决方案可以让我只根据学生的university_id
来连接学生?在
这是我的代码:
import matplotlib.pyplot as plt
import networkx as nx
import MySQLdb
# #############################################################################
# Retrieve the data from remote server.
myDB = MySQLdb.connect(host="*,port=3306,user="mysql",passwd="***")
cHandler = myDB.cursor()
cHandler.execute("USE research_project")
cHandler.execute("SELECT * FROM students")
results = cHandler.fetchall()
G = nx.Graph()
for items in results:
# items[0] is a unique ID, items[1] = university_id, items[2] = full name
G.add_node(items[0], attr_dict={'university_id': items[1], 'full_name': items[2]})
for node_r, attributes in G.nodes(data=True):
key_set = set(attributes.keys())
G.add_edges_from([(node_r, node) for node, attributes in G.nodes(data=True)
if key_set.intersection(set(attributes.keys()))
and node != node_r])
nx.draw(G)
plt.show()
我在小数据集上测试了上面的方法,结果似乎很管用。 我有一种预感,发生的事情与我向节点添加属性的方式有关。在
上面的解决方案需要注意的是,它在运行时非常慢。只要我能想出更快的解决方案,我就会更新我的答案。在
相关问题 更多 >
编程相关推荐