阅读二分图

时间:2015-11-04 04:50:40

标签: python graph networkx

我试图从具有以下结构的.txt文件中读取图表:

115564928125997351000, ah.creativecodeapps.tiempo,1
117923818995099650007, air.com.agg.popcornmakermarket,-1
104000841215686444437, air.com.zahdoo.cadie,1
.
.
.

我一直在使用以下命令:

g=nx.read_weighted_edgelist('siatoy.txt', delimiter=',',nodetype=str,encoding='utf-8')

但是当我g.edges(data=True)时,我得到了这个:

[('106784557494786869271', ' com.map2app.U5635321228165120A5661458385862656'),
('106784557494786869271',' com.jb.gokeyboard.theme.mzdevelopment.americankeyboard'),
('106784557494786869271', ' com.benbasha.whoopeecushion'),
(' com.airplaneflighttakeoff', '115981152169430603941'),...]

但我希望始终将数字id作为元组的第一个元素。请注意,这不会发生在我在示例中显示的列表的最后一个元素中。

我该怎么办?我需要稍后迭代边缘,我需要考虑边缘的顺序,这意味着我需要元组的第一个元素始终是数字ID。

问题是如何在阅读图表或完成图表后实现这一目标?

1 个答案:

答案 0 :(得分:1)

一个想法是使用str.isdigit来测试节点是否为数字。 (例如,请参阅this SO answer。)然后,您可以创建边缘列表,对每个边缘进行排序,以便数字节点排在第一位:

edges = []
for u, v, d in G.edges(data=True): # note that d contains each edge's data
    if u.isdigit(): # if u is numeric put it first
        edges.append( (u, v, d) )
    else:
        edges.append( (v, u, d) )

或以单行形式:

edges = [ (u, v, d) if u.isdigit() else (v, u, d)  for u, v, d in G.edges(data=True) ]
print edges

输出:

[('117923818995099650007', ' air.com.agg.popcornmakermarket', {'weight': -1.0}),
('104000841215686444437', ' air.com.zahdoo.cadie', {'weight': 1.0}),
('115564928125997351000', ' ah.creativecodeapps.tiempo', {'weight': 1.0})]