Changeset 613

Timestamp:

09/11/16 03:59:38 (8 years ago)

Author:

Maciej Komosinski

Message:

Introduced options to flip (invert, negate) X and/or Y axes

File:

• mds-and-trees/mds_plot.py (modified) (5 diffs)

mds-and-trees/mds_plot.py

@@ -103 +103 @@
 
 
-def plot(coordinates, labels, dimensions, variance_fraction, jitter=0, outname=""):
+def plot(coordinates, labels, dimensions, variance_fraction, jitter, flipX, flipY, outname=""):
         fig = plt.figure()
 
@@ -111 +111 @@
                 ax = fig.add_subplot(111, projection='3d')
 
-        add_jitter = lambda tab : rand_jitter(tab, jitter) if jitter>0 else tab
-
         x_dim = len(coordinates[0])
         y_dim = len(coordinates)
 
+        if flipX:
+                coordinates=np.hstack((-coordinates[:, [0]], coordinates[:, [1]]))
+        if flipY:
+                coordinates=np.hstack((coordinates[:, [0]], -coordinates[:, [1]]))
+
+        add_jitter = lambda tab: rand_jitter(tab, jitter) if jitter>0 else tab
         points = [add_jitter(coordinates[:, i]) for i in range(x_dim)]
         
         if labels is not None and dimensions==2: #could be ported to 3D too
                 ax.scatter(*points, alpha=0) #invisible points, because we will show labels instead
-                labelconvert={'velland':'V','velwat':'W','vpp':'P','vpa':'A'} #use this if you want to replace long names with short IDs
-                colors={'velland':'green','velwat':'blue','vpp':'red','vpa':'violet'}
+                labelconvert={'velland_':'V','velwat_':'W','vpp_':'P','vpa_':'A'} #use this if you want to replace long names with short IDs
+                colors={'velland_':'green','velwat_':'blue','vpp_':'red','vpa_':'violet'}
                 #for point in points:
                 #       print(point)
@@ -154 +158 @@
 
 
-def main(filename, dimensions=3, outname="", jitter=0, separator='\t'):
+def main(filename, dimensions=3, outname="", jitter=0, separator='\t', flipX=False, flipY=False):
         distances,labels = read_file(filename, separator)
         embed,variance_fraction = compute_mds(distances, dimensions)
@@ -161 +165 @@
                 embed = np.array([np.insert(e, 0, 0, axis=0) for e in embed])
         
-        plot(embed, labels, dimensions, variance_fraction, jitter, outname)
+        plot(embed, labels, dimensions, variance_fraction, jitter, flipX, flipY, outname)
 
 
@@ -171 +175 @@
         parser.add_argument('--sep', required=False, help='separator of the source file')
         parser.add_argument('--j', required=False, help='for j>0, random jitter is added to points in the plot')
+        parser.add_argument('--flipX', required=False, dest='flipX', action='store_true')
+        parser.add_argument('--flipY', required=False, dest='flipY', action='store_true')
+        parser.set_defaults(flipX=False)
+        parser.set_defaults(flipY=False)
 
         args = parser.parse_args()
-        set_value = lambda value, default : default if value == None else value
-        main(args.input, int(set_value(args.dim, 3)), set_value(args.output, ""), float(set_value(args.j, 0)), set_value(args.sep, "\t"))
+        set_value = lambda value, default: default if value == None else value
+        main(args.input, int(set_value(args.dim, 3)), set_value(args.output, ""), float(set_value(args.j, 0)), set_value(args.sep, "\t"), args.flipX, args.flipY)