创建从EdgeList都邻居列表（Python的：访问元组与指定的第一值）

Question

使用NetworkX，我已经获得描述在图中的边（对顶点的）元组的列表：

G = [(1, 2), (1, 3), (1, 4), (2, 4), (3, 8), (4, 5), (8, 15), (5, 6), (5, 7), (6, 17), (7, 11), (7, 15), (7, 16), (17, 12), (11, 12), (11, 13), (15, 9), (15, 10), (16, 9), (9, 18), (18, 13), (18, 14), (10, 14)] 

使用此列表，我想遍历每个顶点，并找到每个相邻的顶点，但我想按顺序执行此操作。所以我想得到的是例如一个嵌套列表，其中ith子列表包含顶点i的每个邻居。

Neighbors = [[2, 3, 4], [1, 4], [1, 8], [1, 2, 5], [4, 6, 7], [5, 17], [5, 11, 15, 16], [3, 15], [15, 16, 18], [14, 15], [7, 12, 13], [11, 17], [11, 18], [10, 18], [7, 8, 9, 10], [7, 9], [6, 12], [9, 13, 14]] 

，但它也可以是另一个排序的数据结构。但是，由于我的图可能包含一百万条边和顶点，因此我希望实现一个不会遍历每个顶点的整个列表的例程，因为我想保持运行时较低。

有什么办法可以达到这个目的吗？很感谢任何形式的帮助。

Answer 1

可以使用defaultdict如下：

from collections import defaultdict 
d = defaultdict(set) 

for x, y in G: 
    d[x].add(y) 
    d[y].add(x) 

d 
defaultdict(set, 
      {1: {2, 3, 4}, 
      2: {1, 4}, 
      3: {1, 8}, 
      4: {1, 2, 5}, 
      5: {4, 6, 7}, 
      6: {5, 17}, 
      7: {5, 11, 15, 16}, 
      8: {3, 15}, 
      9: {15, 16, 18}, 
      10: {14, 15}, 
      11: {7, 12, 13}, 
      12: {11, 17}, 
      13: {11, 18}, 
      14: {10, 18}, 
      15: {7, 8, 9, 10}, 
      16: {7, 9}, 
      17: {6, 12}, 
      18: {9, 13, 14}}) 

您可以转换字典到列表：

[sorted(d[k]) for k in range(1, max(d.keys())+1)] 
[[2, 3, 4], 
[1, 4], 
[1, 8], 
[1, 2, 5], 
[4, 6, 7], 
[5, 17], 
[5, 11, 15, 16], 
[3, 15], 
[15, 16, 18], 
[14, 15], 
[7, 12, 13], 
[11, 17], 
[11, 18], 
[10, 18], 
[7, 8, 9, 10], 
[7, 9], 
[6, 12], 
[9, 13, 14]] 

Answer 2

如果您有访问原始图形g，您可以通过迭代原始边缘列表（如果不是，您可以从G构造g）：

[list(v.keys()) for _,e in g.edge.items()] 
#[[2, 3, 4], [1, 4], [8, 1], [1, 2, 5], [4, 6, 7], [17, 5], [16, 11, 5, 15], ...] 

可以省略list()如果您没有问题dict_keys，这是一个简单的列表的只读版本：

[v.keys() for _,e in g.edge.items()] 
#[dict_keys([2, 3, 4]), dict_keys([1, 4]), dict_keys([8, 1]), ...]