add shuffle option to args

Yucheng-Zhang · Mar 7, 2019 · a0a2368 · a0a2368
1 parent 5bd6761
commit a0a2368
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/jack.py b/jack.py
@@ -24,6 +24,8 @@
                         help='Jackknife regions will be a Healpix map.')
     parser.add_argument('-plotmap', type=int, default=1,
                         help='Plot jackknife healpix map.')
+    parser.add_argument('-sf', type=int, default=0,
+                        help='Shuffle(1) or not(0) when plotting.')
 
     parser.add_argument('-fmap', type=str, default='',
                         help='Output jackknife regions in Healpix map.')
@@ -71,7 +73,7 @@
 
         if args.plotmap:
             print('-- note: not labeled yet, just demo of regions')
-            miscfuncs.plot_jk_map(jk_map, shuffle=True, njr=args.njr)
+            miscfuncs.plot_jk_map(jk_map, shuffle=args.sf, njr=args.njr)
 
         if args.fbounds != '':
             miscfuncs.save_jk_bounds(jk_bounds, args.fbounds)
@@ -104,4 +106,4 @@
     # analyze labeled random points
     if args.rand_lbed != '':
         rand = miscfuncs.load_data_pd(args.rand_lbed)
-        miscfuncs.analyze_rand(rand)
+        miscfuncs.analyze_rand(rand, args.sf)
diff --git a/miscfuncs.py b/miscfuncs.py
@@ -81,7 +81,7 @@ def rand2map(rand, nside):
     return jk_map
 
 
-def analyze_rand(rand):
+def analyze_rand(rand, sf):
     '''Analyze the labeled random points.
     Columns in [0:RA, 1:DEC, 2:Z, 3:weight, 4:jackknife label].'''
 
@@ -106,4 +106,4 @@ def analyze_rand(rand):
     print(pcdev)
 
     jk_map = rand2map(rand, 256)
-    plot_jk_map(jk_map, shuffle=True, njr=njr)
+    plot_jk_map(jk_map, shuffle=sf, njr=njr)