Weighted fields

compute_mca.py

@@ -5,7 +5,11 @@ import matplotlib.pyplot as plt
 ts = np.fromfile('ts.std', dtype=np.float32).reshape(1147, 28, 31)
 ps = np.fromfile('ps.std', dtype=np.float32).reshape(1147, 28, 31)
 
-svd = supersvd(ts, ps, 4)
+w = np.zeros((28, 31), dtype=np.float32)
+lat = np.deg2rad(35 + 1.5 * np.arange(31))
+w[:, :] = np.cos(lat).reshape(1, -1)
+
+svd = supersvd(ts, ps, 4, WX=w, WY=w)
 
 first_mode_ts = svd.x_vect[0]