networkx代码python的解释

时间:2012-04-06 10:22:10

标签: python networkx

我是编码python的新手。我希望修改此代码以开发一个二分双模式版本。它是来自networkx的代码,用于制作几何随机图。我必须掌握这个功能的大部分内容,但我很难理解94到99行的确切内容。我理解的是,zip和nodes.pop(),但其他部分让新手感到困惑。任何人都可以帮助解释代码的这一部分所做的更多,而不是给出的一般#描述吗?

G=nx.Graph()
G.name="Random Geometric Graph"
G.add_nodes_from(range(n)) 
if pos is None:
    # random positions
    for n in G:
        G.node[n]['pos']=[random.random() for i in range(0,dim)]
else:
    nx.set_node_attributes(G,'pos',pos) 
# connect nodes within "radius" of each other
# n^2 algorithm, could use a k-d tree implementation
nodes = G.nodes(data=True)
while nodes:             #line94
    u,du = nodes.pop()
    pu = du['pos']
    for v,dv in nodes:
        pv = dv['pos']
        d = sum(((a-b)**2 for a,b in zip(pu,pv))) #line99
        if d <= radius**2:
            G.add_edge(u,v)
return G

1 个答案:

答案 0 :(得分:2)

nodes = [some list]
while nodes:
  a = nodes.pop()
  for b in nodes:
    # do something

这段代码是一个非常常见的习惯用法,它将每个节点与每个其他节点完全一次组合在一起(因此ab的顺序无关紧要对于# do something部分中执行的操作。

它的工作原理是因为空列表在while循环的条件下被视为假值,而非空列表被认为是布尔值真。

d = sum(((a-b)**2 for a,b in zip(pu,pv)))

此行计算两个向量pupv的{​​{3}}的平方。将它拆开可以最好地证明这一点:

>>> pu = (6,6,6)
>>> pv = (1,3,7)
>>> zip(pu, pv)
[(6, 1), (6, 3), (6, 7)]
>>> [(a-b) for a,b in zip(pu, pv)]
[5, 3, -1]
>>> [(a-b)**2 for a,b in zip(pu, pv)]
[25, 9, 1]
>>> sum((a-b)**2 for a,b in zip(pu, pv))
35

在最后一步中,我们不再使用列表推导,因为我们不需要列表。 sum只需要某种可迭代形式的值,因此我们使用生成器表达式。