NetworkX / Pandas - 如何将每个节点所在的社区组输出为.txt文件

Question

我需要将网络中每个节点的社区输出到一个 .txt 文件中。我正在使用 NetworkX ver. 2.1 和 Pandas ver. 0.23.4：

import networkx as nx
import pandas as pd
from networkx.algorithms import community

G = nx.barbell_graph(5, 1)
communities_generator = community.girvan_newman(G)
top_level_communities = next(communities_generator)
next_level_communities = next(communities_generator)
sorted(map(sorted, next_level_communities))

#>>> sorted(map(sorted, next_level_communities))
[[0, 1, 2, 3, 4], [5], [6, 7, 8, 9, 10]]
#in this case, 3 different communities (groups) were identified

我需要一个类似于以下格式的 .txt 表格：

NODE    COMMUNITY
0   GROUP 1
1   GROUP 1
2   GROUP 1
3   GROUP 1
4   GROUP 1
5   GROUP 2
6   GROUP 3
7   GROUP 3
8   GROUP 3
9   GROUP 3
10  GROUP 3

- ASF

Answer 1

您可以这样做：

import networkx as nx
import pandas as pd
from networkx.algorithms import community

G = nx.barbell_graph(5, 1)
communities_generator = community.girvan_newman(G)
top_level_communities = next(communities_generator)
next_level_communities = next(communities_generator)

data = [[element, "GROUP-{}".format(ii + 1)] for ii, st in enumerate(next_level_communities) for element in sorted(st)]
print(data)

frame = pd.DataFrame(data=data, columns=['node', 'community'])
frame.to_csv("communities.csv", sep=" ", index=False)

文件'communities.csv'的格式如下：

node community
0 GROUP-1
1 GROUP-1
2 GROUP-1
3 GROUP-1
4 GROUP-1
5 GROUP-2
6 GROUP-3
7 GROUP-3
8 GROUP-3
9 GROUP-3
10 GROUP-3

- Dani Mesejo