/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

1 vote305 views1 answer
Elif in django template
Python
122 votes288 views3 answers
Iterate through keys in dictionary
Python
219 votes664 views6 answers
Random number python
Python
66 votes315 views6 answers
Python delete folder and contents
Python
3 votes331 views2 answers
Change django administration title
Python Django
0 vote308 views1 answer
Couchbase python
Python
14 votes301 views6 answers
Python binary search
Python
1 vote329 views3 answers
Create spark dataframe from pandas
Python
112 votes254 views10 answers
December global festivities
Python
3 votes313 views1 answer
How to set index pandas
Python