diff --git a/facebook/gephi.py b/facebook/gephi.py index 88c8487..5c61b81 100644 --- a/facebook/gephi.py +++ b/facebook/gephi.py @@ -120,7 +120,7 @@ def export_network(member, n=2): member_friends = set(member_friends) members = [] for key, value in d.items(): - if len(value & member_friends) >= n: + if len(value & member_friends) >= n or key in member_friends: members.append(key) # Skapa df med edges @@ -162,7 +162,7 @@ def export_network(member, n=2): edges, source="_from", target="_to", - edge_attr=["reaction", "date", "now"], + edge_attr=["reaction", "date"], #, "now" create_using=nx.MultiDiGraph, ) @@ -177,5 +177,5 @@ def export_network(member, n=2): if __name__ == "__main__": - export_network("maria.hansson.botin") + export_network("asifasghar") # export_network(input('Member: '))