Создать список ребер из dataframe с группами в Python - PullRequest
Новая
       98

Создать список ребер из dataframe с группами в Python

0 голосов
/ 14 ноября 2018

У меня есть следующие данные: 

import pandas as pd
employee=["a","b","a","c","d","e","c","d","f"]
project=[1,1,2,2,2,3,3,4,4]
df=pd.DataFrame({"employee":employee,
             "project":project})

Я хочу создать список ребер из этого кадра данных.В прошлом, когда я использовал RI, использовался следующий код: 

edges<-unique(df %>% group_by(project) %>%
            filter(n()>=2) %>% group_by(project) %>%
            do(data.frame(t(combn(.$employee, 2)), stringsAsFactors=FALSE)))
edges<-subset(edges,as.numeric(edges$X1)-as.numeric(edges$X2)!=0)

Однако у меня не получается, когда я хочу сделать то же самое в Python.Кто-нибудь может помочь с некоторыми советами о том, как преобразовать это (возможно, с помощью матрицы смежности) в список краев.

Желаемый результат должен выглядеть примерно так: 

employee1  employee2
A            B 
A            C
C            D  
E            C
D            F

РЕДАКТИРОВАТЬ: Iнаконец-то нашел мой ответ: pandas - преобразовать фрейм данных в список краев в соответствии со значениями столбцов

...