我试图从具有以下结构的.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。
问题是如何在阅读图表或完成图表后实现这一目标?
答案 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})]