(libhookecurve.py, hooke.py , fit.py, generalvclamp.py) Added color attribute to...

[hooke.git] / pcluster.py

diff --git a/pcluster.py b/pcluster.py
index 583edad306525aaa02b611270c97d17a2f8a17a6..8f144fcd787c5d31fccc05e52f5f6d0e21104896 100644 (file)
--- a/pcluster.py
+++ b/pcluster.py
@@ -325,6 +325,7 @@ class pclusterCommands:
                 if (row[0]<9000 and row[1]<9000 and row[2]<9000 and row[3]<9000 and row[4]<9000 and row[5]<9000):
                     if (row[0]>0 and row[1]>0 and row[2]>0 and row[3]>0 and row[4]>0 and row[5]>0):
                         self.pca_paths[nPlotGood] = plot_path_temp
+                        row=[row[1],row[2]]
                         self.pca_myArray.append(row)
                         nPlotGood = nPlotGood+1
                         
@@ -348,10 +349,24 @@ class pclusterCommands:
         X=myArrayTr[0]
         Y=myArrayTr[1]
         clustplot=lhc.PlotObject()
-        clustplot.add_set(X,Y)
-        #clustplot.add_set(X[:14],Y[:14])
+        
+        Xsyn=[]
+        Ysyn=[]
+        Xgb1=[]
+        Ygb1=[]
+        for index in range(len(self.pca_paths)):
+            if 'syn' in self.pca_paths[index]:
+                Xsyn.append(X[index])
+                Ysyn.append(Y[index])
+            else:
+                Xgb1.append(X[index])
+                Ygb1.append(Y[index])
+        
+        clustplot.add_set(Xsyn,Ysyn)
+        clustplot.add_set(Xgb1,Ygb1)
         clustplot.normalize_vectors()
-        clustplot.styles=['scatter']
+        clustplot.styles=['scatter', 'scatter']
+        clustplot.colors=[None,'red']
         #clustplot.styles=['scatter',None]
         clustplot.destination=1
         self._send_plot([clustplot])