Changeset 616 for mds-and-trees

Timestamp:

09/15/16 00:07:27 (8 years ago)

Author:

Maciej Komosinski

Message:

Introduced an option to swap (exchange) X and 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, flipX, flipY, outname=""):
+def plot(coordinates, labels, dimensions, variance_fraction, jitter, flipX, flipY, swapXY, outname=""):
         fig = plt.figure()
 
@@ -118 +118 @@
         if flipY:
                 coordinates=np.hstack((coordinates[:, [0]], -coordinates[:, [1]]))
+        if swapXY:
+                coordinates[:,[0, 1]] = coordinates[:,[1, 0]]
 
         add_jitter = lambda tab: rand_jitter(tab, jitter) if jitter>0 else tab
@@ -158 +160 @@
 
 
-def main(filename, dimensions=3, outname="", jitter=0, separator='\t', flipX=False, flipY=False):
+def main(filename, dimensions=3, outname="", jitter=0, separator='\t', flipX=False, flipY=False, swapXY=False):
         distances,labels = read_file(filename, separator)
         embed,variance_fraction = compute_mds(distances, dimensions)
@@ -165 +167 @@
                 embed = np.array([np.insert(e, 0, 0, axis=0) for e in embed])
         
-        plot(embed, labels, dimensions, variance_fraction, jitter, flipX, flipY, outname)
+        plot(embed, labels, dimensions, variance_fraction, jitter, flipX, flipY, swapXY, outname)
 
 
@@ -177 +179 @@
         parser.add_argument('--flipX', required=False, dest='flipX', action='store_true')
         parser.add_argument('--flipY', required=False, dest='flipY', action='store_true')
+        parser.add_argument('--swapXY', required=False, dest='swapXY', action='store_true')
         parser.set_defaults(flipX=False)
         parser.set_defaults(flipY=False)
+        parser.set_defaults(swapXY=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"), args.flipX, args.flipY)
+        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, args.swapXY)