markovmodel / PyEMMA

🚂 Python API for Emma's Markov Model Algorithms 🚂
http://pyemma.org
GNU Lesser General Public License v3.0
306 stars 119 forks source link

pyemma.plots.scatter_contour produce an error #1606

Open hima111997 opened 1 year ago

hima111997 commented 1 year ago

path = 'path is here' top = 'psf top file' os.chdir(path) feat = pyemma.coordinates.featurizer(top)

feat.add_backbone_torsions(periodic=False, selstr='index 0 to 6090') trr = 'trr file with 4001 frame' data = pyemma.coordinates.source(trr , features=feat) lag = 10 tica = pyemma.coordinates.tica(data, dim=2, lag=lag, kinetic_map=False) tica_Y = tica.get_output()[0] cl = pyemma.coordinates.cluster_kmeans(data= tica_Y, k=200, stride=2, max_iter=40) cl_dtrajs = cl.dtrajs cc_x = cl.clustercenters[:,0] cc_y = cl.clustercenters[:,1] M = msm.estimate_markov_model(cl_dtrajs, lag=lag) rn = M.eigenvectors_right()[:,3] # second motion ax = mplt.scatter_contour(cc_x, cc_y, rn, cmap='seismic') # <-- this produces an error ( the number of elements in cc_x and cc_y is smaller than rn variable)