read the edge list from a csv file and create a graph with networkx

Question

Hi everyone I want to read the edge list from a csv file and create a graph with networkx to calculate the betweenness centrality with python. My code is:

import pandas as pd
import networkx as nx
df = pd.read_csv('edges1.csv')
Graphtype = nx.Graph()
G = nx.from_pandas_edgelist(df, edge_attr='genre_ids', create_using=Graphtype)
centrality = nx.betweenness_centrality(G, normalize=False)
print(centrality)

edges1.csv have 97180 row:

Surce,Target,genre_ids
Avatar,Violent Night,18
Harry Potter,The Woman King,20
Happy Feet, Froze,23
so on....

My code give me error: KeyError: 'source'. How can i do?

I didn't test it but I guess instead of Surce,Target,genre_ids in the csv you should put source,target,genre_ids, so with an o in source and lower case. — Marijn
– Marijn, Commented Jan 13, 2023 at 19:47
from_pandas_edgelist requires valid source column, fix your file header — RomanPerekhrest
– RomanPerekhrest, Commented Jan 13, 2023 at 19:49

SultanOrazbayev · Accepted Answer · 2023-01-14 02:07:27Z

1

When loading the data, make sure the column names in csv file match the default expected values OR specify the custom names.

If the column names are "Surce,Target,genre_ids" (as in the snippet provided by OP), then the appropriate command is:

G = nx.from_pandas_edgelist(
    df,
    source="Surce",
    target="Target",
    edge_attr='genre_ids',
    create_using=Graphtype
)

See the docs.

answered Jan 14, 2023 at 2:07

SultanOrazbayev

16.7k3 gold badges25 silver badges59 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

read the edge list from a csv file and create a graph with networkx

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related