gexf export colors2

Author: alexrd

csnanalysis/csn.py

@@ -276,13 +276,14 @@ def calc_committors(self,basins,labels=None,basin_labels=None,add_basins=False,t
         else:
             # use trimmed transition matrix
             comm = committor(self.transmat,basins,tol=tol,maxstep=maxstep)
-            full_comm = np.zeros(self.nnodes,dtype=float64)
+            full_comm = np.zeros((self.transmat.shape[0],len(basins)),dtype=float)
             for i,ind in enumerate(self.trim_indices):
                 full_comm[ind] = comm[i]
 
         if labels is None:
             labels = ['p' + str(i) for i in range(len(basins))]
-        for i,b in enumerate(basins):
+
+        for i in range(len(basins)):
             fc_float = [float(i) for i in full_comm[:,i]]
             self.add_attr(labels[i], fc_float)
 

csnanalysis/matrix.py

@@ -137,6 +137,7 @@ def committor(transmat,basins,tol=1e-6,maxstep=20):
     """
 
     # make sink_matrix
+
     flat_sink = [i for b in basins for i in b]
     sink_mat = make_sink(transmat,flat_sink)