看看这个:
df = pd.Dataframe({'node': ['a', 'b','c', 'd', 'e', 'f', 'g'], 't1': ['pos', 'neg', 'neg', 'pos', 'neg', 'pos', 'neg'], 't2': ['neg', 'neg', 'neg', 'neg', 'pos', 'neg', 'pos']})K = nx.Graph()K.add_nodes_from(df['node'].values)# Create edgesfor i, group in df.groupby(['t1', 't2'])['node']: # generate all combinations without replacement # from the group of similar column pairs for u, v in itertools.combinations(group, 2): K.add_edge(u, v)print(K.edges())结果:[(’a’,’d’),(’a’,’f’),(’c’,’b’),(’e’,’g’),(’d’,’f ‘)]
这里的技巧是在熊猫中同时将2列分组。然后,您可以只创建要添加到图中的所有边的组合。



