/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

3 votes109 views1 answer
Restore index after concatenate
Python
1 vote148 views1 answer
Django gmail smtp
Python
0 vote107 views1 answer
Conversion of int to a specified base number
Python
2 votes150 views1 answer
Difference between w+ and r+ in python
Python
0 vote109 views1 answer
Reload is not defined python 3
Python
0 vote108 views1 answer
Python API translate language into Igbo
Python
24 votes67 views4 answers
Valueerror: the truth value of a series is ambiguous. use a.empty, a.bool(), a.item(), a.any() or a.all().
Python
0 vote183 views4 answers
"%(class)s" in django
Python Django
5 votes122 views2 answers
Append string variable with integer python
Python
0 vote119 views1 answer
Change every element of list python with map
Python