从csv-File中创建networkx中的边

时间:2015-10-12 13:15:32

标签: python csv for-loop row networkx

我从事经济学工作,而且我是编程方面的新手,所以请赦免我缺乏技术知识。

m using iPython 2.7 and I want to analyze a production system as network. Therefore I尝试用csv文件创建网络。 csv文件包含两列。第一列是工作订单的表示,第二列是工作站。工作站必须是网络的节点。

ORDER_ID;MACHINE_ID;
0;0;
0;1;
1;1;
2;2;
2;3;
2;4;
2;5;
2;6;
2;1;
2;7;
2;2;
3;8;
3;1;
4;9;
5;10;
5;10;
5;5;
5;11;
5;0; 
5;12;

只要订单ID在不同行中的编号相同,这些行的节点就必须作为边链接。订单ID更改后,下一行必须检查并链接为边缘。边缘用于更好的分析。我写了一个算法,但是我在挑选行项时遇到了问题。

import networkx as nx
import csv 

g = nx.DiGraph()

with open("Data.csv") as dataSet:
    data = csv.reader()
    for OrderID1 in dataSet:
        OrderID = OrderID[0]
        for MachineID1 in dataSet:
            MachineID = MachineID1[1]
                if OrderID[0][i] == OrderID[0][i+1]:  
                    g.add_edge(MachineID[0][i],MachineID[0][i+1])
                elif OrderID[0][i] != row[0][i+1]: 
                    g.add_edge(row[0][i+1],row[0][i+2])
                else:
                    dataSet.close()

结果必须如下所示:

g.edges()

>>> (0,1),(2,3),(3,4),(4,5),(5,6),(6,1),(1,7),(7,2),(8,1),(10,10),(10,5),(5,11),(11,0),(0,12)

非常感谢每个需要时间回应的人!

1 个答案:

答案 0 :(得分:2)

我认为算法应如下:

import csv
import networkx as nx
g = nx.DiGraph()
data_csv = open('Data.csv')
csv_f = csv.reader(data_csv,delimiter=';')
prev_orderID = -1
prev_machineID = -1
for row in csv_f:
    if row[0] == prev_orderID:
        g.add_edge(prev_machineID,row[1])
    prev_orderID = row[0]
    prev_machineID = row[1]
print g.edges()

我已将您提供的样本放入.csv文件中,结果与您的期望相同:

[('11', '0'), ('10', '10'), ('10', '5'), ('1', '7'), ('0', '1'), ('0', '12'), ('3', '4'), ('2', '3'), ('5', '11'), ('5', '6'), ('4', '5'), ('7', '2'), ('6', '1'), ('8', '1')]
  

也许您在阅读行时遇到问题,因为您没有指定分隔符。