/Python/Networkx node attribute from a dataframe

Networkx node attribute from a dataframe

import networkx as nx
import pandas as pd

# Build a sample dataframe (with 2 edges: 0 -> 1, 0 -> 2, node 0 has attr_1 value of 'a', node 1 has 'b', node 2 has 'c')
d = {'node_from': [0, 0], 'node_to': [1, 2], 'src_attr_1': ['a','a'], 'tgt_attr_1': ['b', 'c']}
df = pd.DataFrame(data=d)
G = nx.from_pandas_edgelist(df, 'node_from', 'node_to')

# Iterate over df rows and set the source and target nodes' attributes for each row:
for index, row in df.iterrows():
    G.nodes[row['node_from']]['attr_1'] = row['src_attr_1']
    G.nodes[row['node_to']]['attr_1'] = row['tgt_attr_1']

print(G.edges())
print(G.nodes(data=True))

Source: stackoverflow.com

Similar codes

5 votes332 views4 answers
How print 2 decimal in python
Python
28 votes360 views5 answers
String slicing in python
Python
105 votes394 views2 answers
Create react-app
Python
0 vote409 views1 answer
Python trim zero off end of list
Python
0 vote376 views1 answer
Celery beat command for django celery beat
Python Django
0 vote293 views1 answer
Fill in the gaps in the initials function so that it returns the initials of the words contained in the phrase received, in upper case.
Python
19 votes345 views3 answers
Python program to check whether a number is even or odd
Python
0 vote399 views1 answer
Python instagram login
Python
3 votes343 views1 answer
Local testing server Python
Python
0 vote341 views1 answer
Machine learning project outline
Python