Convert adjacency matrix to adjacency list

adjacency_matrix_to_adjacency_list.py

import numpy as np

def adjacency_matrix_to_adjacency_list(adjacency_matrix):
    adjacency_list = []
    indices = np.argwhere(adjacency_matrix==1)
    
    for node in range(len(adjacency_matrix)):
        adjacency_list.append([])
        for index in indices:
            if index[0] == node:
                adjacency_list[node].append(index[1])
    adjacency_list = [np.asarray(l) for l in adjacency_list]
    return adjacency_list