def test_graph_create_from_coo_matrix(self): """ Test the creation of a graph from a sparse coo_matrix """ import scipy.sparse as spp a = np.random.randn(5, 5)>.8 s = spp.coo_matrix(a) wg = fg.wgraph_from_coo_matrix(s) b = wg.adjacency() self.assert_((a==b).all())
def reduce_coo_matrix(mat, mask): """ reduce a supposedly coo_matrix to the vertices in the mask Parameters ---------- mat: sparse coo_matrix, input matrix mask: boolean array of shape mat.shape[0], desired elements """ G = fg.wgraph_from_coo_matrix(mat) K = G.subgraph(mask) return K.to_coo_matrix()