擅长:python、mysql、java
<p>在@etep和@MarkRansom的帮助下解决了这个问题!我必须打开文件并在整个<code>for-loop</code>之前定义<code>writer</code>。在</p>
<pre><code># open file and define writer
with open('out_file.csv', 'wb') as f:
writer = csv.writer(f)
# data processing
seen = set()
for u in name_nodes:
# seen=set([u]) # print both u-v, and v-u
seen.add(u) # don't print v-u
unbrs = set(B[u])
nbrs2 = set((n for nbr in unbrs for n in B[nbr])) - seen
for v in nbrs2:
vnbrs = set(B[v])
common = unbrs & vnbrs
weight = len(common)
row = u, v, weight
# write row for each record
writer.writerow(row)
</code></pre>