/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

8 votes275 views3 answers
Sklearn roc curve
Python
0 vote305 views1 answer
Django queryset all objects
Python Django
42 votes270 views5 answers
Python open website
Python
0 vote268 views1 answer
Randomly pick a value in the list
Python
0 vote262 views1 answer
Float division in python
Python
26 votes284 views1 answer
Python gmail
Python
95 votes268 views5 answers
Python string in string
Python
13 votes270 views6 answers
Python length
Python
0 vote290 views1 answer
Print the multiple data types in a single program in list data structures
Python
1 vote294 views1 answer
Discord.py on command error
Python