我是编码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
答案 0 :(得分:2)
nodes = [some list]
while nodes:
a = nodes.pop()
for b in nodes:
# do something
这段代码是一个非常常见的习惯用法,它将每个节点与每个其他节点完全一次组合在一起(因此a
和b
的顺序无关紧要对于# do something
部分中执行的操作。
它的工作原理是因为空列表在while
循环的条件下被视为假值,而非空列表被认为是布尔值真。
d = sum(((a-b)**2 for a,b in zip(pu,pv)))
此行计算两个向量pu
和pv
的{{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
只需要某种可迭代形式的值,因此我们使用生成器表达式。