[英]Python: How to create graph nodes and edges from csv file?
我有一个csv数据文件,其中每一行表示一个事件。 一个简化的例子是
Datetime ColA ColB ColC
2015/07/12 08:45:34 ABC 12
2015/07/12 08:46:04 DCD 10 ABC
2015/07/12 08:46:23 XYZ 34 ABC
2015/07/12 08:46:56 MNO 10 XYZ
2015/07/12 08:46:56 FGH 20
因此,每一行将是一个node
,其属性由每一列Datetime, ColA, ColB, ColC
值表示。 这些节点中的每一个都通过ColA
和ColC
之间的关系连接。
因此,在这个示例中,存在从第1行的边缘到行2和行3中,由于ColC
后者行等于ColA
第一行的。 第3行和第4行通过有向边类似地连接。
第1行没有ColC,因此它没有连接到更高的任何节点,第5行也没有连接。
如何创建图形数据结构以在Python中创建此关系? 它们都应按时间顺序排序,并且如果有两行ColA
与一行ColC
相匹配,则选择时间上更接近的ColC
行。
您可以构建一个二部图,其中日期时间为一部分,而ColA / ColC值为另一部分。 然后将图形“投影”到日期时间节点上-如果两个日期时间都链接到ColA / ColC节点,则在两个日期时间之间创建链接。
这是一些代码,显示了创建无向图的一种方法。 我不明白您的示例中的指示含义。
import csv
import StringIO
import networkx as nx
from networkx.algorithms import bipartite
data ="""Datetime,ColA,ColB,ColC
2015/07/12 08:45:34,ABC,12,
2015/07/12 08:46:04,DCD,10,ABC
2015/07/12 08:46:23,XYZ,34,ABC
2015/07/12 08:46:56,MNO,10,XYZ
2015/07/12 08:46:56,FGH,20,"""
G = nx.Graph()
csvfile = StringIO.StringIO(data)
reader = csv.DictReader(csvfile)
nodes = []
for row in reader:
nodes.append(row['Datetime'])
G.add_node(row['Datetime'])
if row['ColA'] != '':
G.add_edge(row['Datetime'],row['ColA'])
if row['ColC'] != '':
G.add_edge(row['Datetime'],row['ColC'])
print G.edges()
B = bipartite.projected_graph(G, nodes)
print B.edges()
OUTPUT
[('2015/07/12 08:46:23', 'XYZ'), ('2015/07/12 08:46:23', 'ABC'), ('ABC', '2015/07/12 08:46:04'), ('ABC', '2015/07/12 08:45:34'), ('DCD', '2015/07/12 08:46:04'), ('FGH', '2015/07/12 08:46:56'), ('2015/07/12 08:46:56', 'XYZ'), ('2015/07/12 08:46:56', 'MNO')]
[('2015/07/12 08:46:23', '2015/07/12 08:46:04'), ('2015/07/12 08:46:23', '2015/07/12 08:46:56'), ('2015/07/12 08:46:23', '2015/07/12 08:45:34'), ('2015/07/12 08:46:04', '2015/07/12 08:45:34')]
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.