top of page

Data Scientists, The 5 Graph Algorithms that you should know

We as data scientists have gotten quite comfortable with Pandas or SQL or any other relational database. We are used to seeing our users in rows with their attributes as columns. But does the real world really behave like that?'

In a connected world, users cannot be considered as independent entities. They have got certain relationships between each other and we would sometimes like to include such relationships while building our machine learning models.

Now while in a relational database, we cannot use such relations between different rows(users), in a graph database it is fairly trivial to do that.

Graph algorithms are a set of instructions that traverse (visits nodes of a) graph. Some algorithms are used to find a specific node or the path between two given nodes.

1. Connected Components

A graph with 3 connected components

We all know how clustering works?

Read more: Clustering

As a concrete example: Say you have data about roads joining any two cities in the world. And you need to find out all the continents in the world and which city they contain.

The connected components algorithm that we use to do this is based on a special case of BFS/DFS.


From a Retail Perspective: Let us say, we have a lot of customers using a lot of accounts. One way in which we can use the Connected components algorithm is to find out distinct families in our dataset.

We can assume edges(roads) between CustomerIDs based on same credit card usage, or same address or same mobile number, etc. Once we have those connections, we can then run the connected component algorithm on the same to create individual clusters to which we can then assign a family ID.

We can then use these family IDs to provide personalized recommendations based on family needs. We can also use this family ID to fuel our classification algorithms by creating grouped features based on family.

From a Finance Perspective: Another use case would be to capture fraud using these family IDs. If an account has done fraud in the past, it is highly probable that the connected accounts are also susceptible to fraud.

The possibilities are only limited by your own imagination.


We will be using the Networkx module in Python for creating and analyzing our graphs.

Let us start with an example graph which we are using for our purpose. Contains cities and distance information between them.

Graph with Some random distances

We first start by creating a list of edges along with the distances which we will add as the weight of the edge:

edgelist = [
    ['Mannheim', 'Frankfurt', 85], 
    ['Mannheim', 'Karlsruhe', 80], 
    ['Erfurt', 'Wurzburg', 186], 
    ['Munchen', 'Numberg', 167], 
    ['Munchen', 'Augsburg', 84], 
    ['Munchen', 'Kassel', 502], 
    ['Numberg', 'Stuttgart', 183],
     ['Numberg', 'Wurzburg', 103], 
     ['Numberg', 'Munchen', 167], 
     ['Stuttgart', 'Numberg', 183], 
     ['Augsburg', 'Munchen', 84], 
     ['Augsburg', 'Karlsruhe', 250], 
     ['Kassel', 'Munchen', 502], 
     ['Kassel', 'Frankfurt', 173], 
     ['Frankfurt', 'Mannheim', 85], 
     ['Frankfurt', 'Wurzburg', 217], 
     ['Frankfurt', 'Kassel', 173], 
     ['Wurzburg', 'Numberg', 103], 
     ['Wurzburg', 'Erfurt', 186], 
     ['Wurzburg', 'Frankfurt', 217], 
     ['Karlsruhe', 'Mannheim', 80], 
     ['Karlsruhe', 'Augsburg', 250],
     ["Mumbai", "Delhi",400],
     ["Delhi", "Kolkata",500],
     ["Kolkata", "Bangalore",600],
     ["TX", "NY",1200],
     ["ALB", "NY",800]]

Let us create a graph using Networkx:

g = nx.Graph() 
for edge in edgelist:     
    g.add_edge(edge[0],edge[1], weight = edge[2])

Now we want to find out distinct continents and their cities from this graph.

We can now do this using the connected components algorithm as:

for i, x in enumerate(nx.connected_components(g)):         
cc0: {
cc1: {
cc2: {

As you can see we are able to find distinct components in our data. Just by using Edges and Vertices. This algorithm could be run on different data to satisfy any use case that I presented above.

2. Shortest Path

Continuing with the above example only, we are given a graph with the cities of Germany and the respective distance between them.

The algorithm that we use for this problem is called Dijkstra. In Dijkstra’s own words:

Dijkstra’s algorithm is very similar to Prim’s algorithm for minimum spanning tree. Like Prim’s MST, we generate a SPT (shortest path tree) with a given source as a root. We maintain two sets, one set contains vertices included in the shortest-path tree, other set includes vertices not yet included in the shortest-path tree. At every step of the algorithm, we find a vertex that is in the other set (set of not yet included) and has a minimum distance from the source.


  • Variations of the Dijkstra algorithm is used extensively in Google Maps to find the shortest routes.

  • You are in a Walmart Store. You have different Aisles and distance between all the aisles. You want to provide the shortest pathway to the customer from Aisle A to Aisle D.

  • You have seen how LinkedIn shows up 1st-degree connections, 2nd-degree connections. What goes on behind the scenes?


print(nx.shortest_path(g, 'Stuttgart','Frankfurt',weight='weight')) print(nx.shortest_path_length(g, 'Stuttgart','Frankfurt',weight='weight')) 
['Stuttgart', 'Numberg', 'Wurzburg', 'Frankfurt'] 

You can also find Shortest paths between all pairs using:

for x in nx.all_pairs_dijkstra_path(g,weight='weight'):     
('Mannheim', {'Mannheim': ['Mannheim'], 'Frankfurt': ['Mannheim', 'Frankfurt'], 'Karlsruhe': ['Mannheim', 'Karlsruhe'], 'Augsburg': ['Mannheim', 'Karlsruhe', 'Augsburg'], 'Kassel': ['Mannheim', 'Frankfurt', 'Kassel'], 'Wurzburg': ['Mannheim', 'Frankfurt', 'Wurzburg'], 'Munchen': ['Mannheim', 'Karlsruhe', 'Augsburg', 'Munchen'], 'Erfurt': ['Mannheim', 'Frankfurt', 'Wurzburg', 'Erfurt'], 'Numberg': ['Mannheim', 'Frankfurt', 'Wurzburg', 'Numberg'], 'Stuttgart': ['Mannheim', 'Frankfurt', 'Wurzburg', 'Numberg', 'Stuttgart']})('Frankfurt', {'Frankfurt': ['Frankfurt'], 'Mannheim': ['Frankfurt', 'Mannheim'], 'Kassel': ['Frankfurt', 'Kassel'], 'Wurzburg': ['Frankfurt', 'Wurzburg'], 'Karlsruhe': ['Frankfurt', 'Mannheim', 'Karlsruhe'], 'Augsburg': ['Frankfurt', 'Mannheim', 'Karlsruhe', 'Augsburg'], 'Munchen': ['Frankfurt', 'Wurzburg', 'Numberg', 'Munchen'], 'Erfurt': ['Frankfurt', 'Wurzburg', 'Erfurt'], 'Numberg': ['Frankfurt', 'Wurzburg', 'Numberg'], 'Stuttgart': ['Frankfurt', 'Wurzburg', 'Numberg', 'Stuttgart']})....

3. Minimum Spanning Tree

Now we have another problem. We work for a water pipe laying company or an internet fiber company. We need to connect all the cities in the graph we have using the minimum amount of wire/pipe.

An Undirected Graph and its MST on the right.


  • Minimum spanning trees have direct applications in the design of networks, including computer networks, telecommunications networks, transportation networks, water supply networks, and electrical grids (which they were first invented for)

  • MST is used for approximating the traveling salesman problem

  • Clustering — First construct MST and then determine a threshold value for breaking some edges in the MST using Intercluster distances and Intracluster distances.

  • Image Segmentation — It was used for Image segmentation where we first construct an MST on a graph where pixels are nodes and distances between pixels are based on some similarity measure(color, intensity, etc.)


# nx.minimum_spanning_tree(g) returns a instance of type graph nx.draw_networkx(nx.minimum_spanning_tree(g))

The MST of our graph.

As you can see the above is the wire we gotta lay.

4. Pagerank

This is the page sorting algorithm that powered google for a long time. It assigns scores to pages based on the number and quality of incoming and outgoing links.


Pagerank can be used anywhere where we want to estimate node importance in any network.

  • It has been used for finding the most influential papers using citations.

  • Has been used by Google to rank pages

  • It can be used to rank tweets- User and Tweets as nodes. Create Link between user if user A follows user B and Link between user and Tweets if user tweets/retweets a tweet.

  • Recommendation engines


For this exercise, we are going to be using Facebook data. We have a file of edges/links between facebook users. We first create the FB graph using:

# reading the datasetfb = nx.read_edgelist('../input/facebook-combined.txt', create_using = nx.Graph(), nodetype = int)

This is how it looks:

pos = nx.spring_layout(fb)import warnings warnings.filterwarnings('ignore')'fivethirtyeight') plt.rcParams['figure.figsize'] = (20, 15) 
nx.draw_networkx(fb, pos, with_labels = False, node_size = 35)

FB User Graph

Now we want to find the users having high influence capability.

Intuitively, the Pagerank algorithm will give a higher score to a user who has a lot of friends who in turn have a lot of FB Friends.

pageranks = nx.pagerank(fb) 
{0: 0.006289602618466542,  
1: 0.00023590202311540972,  
2: 0.00020310565091694562,  
3: 0.00022552359869430617,  
4: 0.00023849264701222462, 

We can get the sorted PageRank or most influential users using:

import operator 
sorted_pagerank = sorted(pagerank.items(), key=operator.itemgetter(1),reverse = True) print(sorted_pagerank) 
[(3437, 0.007614586844749603), 
(107, 0.006936420955866114), 
(1684, 0.0063671621383068295), 
(0, 0.006289602618466542), 
(1912, 0.0038769716008844974), 
(348, 0.0023480969727805783), 
(686, 0.0022193592598000193), 
(3980, 0.002170323579009993), 
(414, 0.0018002990470702262), 
(698, 0.0013171153138368807), 
(483, 0.0012974283300616082), 
(3830, 0.0011844348977671688), 
(376, 0.0009014073664792464), 
(2047, 0.000841029154597401), 
(56, 0.0008039024292749443), 
(25, 0.000800412660519768), 
(828, 0.0007886905420662135), 
(322, 0.0007867992190291396),......

The above IDs are for the most influential users.

We can see the subgraph for the most influential user:

first_degree_connected_nodes = list(fb.neighbors(3437)) second_degree_connected_nodes = [] 
for x in first_degree_connected_nodes:     
    second_degree_connected_nodes+=list(fb.neighbors(x)) second_degree_connected_nodes.remove(3437) 
second_degree_connected_nodes = list(set(second_degree_connected_nodes))
    subgraph_3437 = nx.subgraph
    pos = nx.spring_layout(subgraph_3437)
    node_color = ['yellow' if v == 3437 else 'red' for v in subgraph_3437] 
node_size =  [1000 if v == 3437 else 35 for v in subgraph_3437]'fivethirtyeight') 
plt.rcParams['figure.figsize'] = (20, 15) 
plt.axis('off')nx.draw_networkx(subgraph_3437, pos, with_labels = False, node_color=node_color,node_size=node_size )

Our most influential user(Yellow)

5. Centrality Measures

There are a lot of centrality measures which you can use as features to your machine learning models. I will talk about two of them. You can look at other measures here.

Betweenness Centrality: It is not only the users who have the most friends that are important, the users who connect one geography to another are also important as that lets users see content from diverse geographies. Betweenness centrality quantifies how many times a particular node comes in the shortest chosen path between two other nodes.

Degree Centrality: It is simply the number of connections for a node.


Centrality measures can be used as a feature in any machine learning model.


Here is the code for finding the Betweenness centrality for the subgraph.

pos = nx.spring_layout(subgraph_3437) 
betweennessCentrality = nx.betweenness_centrality(subgraph_3437,normalized=True, endpoints=True)node_size =  [v * 10000 for v in betweennessCentrality.values()] 
nx.draw_networkx(subgraph_3437, pos=pos, with_labels=False,                  node_size=node_size ) 

You can see the nodes sized by their betweenness centrality values here. They can be thought of as information passers. Breaking any of the nodes with a high betweenness Centrality will break the graph into many parts.

Resource: KDNuggets

The Tech Platform

Recent Posts

See All


bottom of page