From 1a95c186664dfda0843a49bbe5ba56ea02bff14d Mon Sep 17 00:00:00 2001
From: Lionel GUEZ <guez@lmd.ens.fr>
Date: Thu, 4 Mar 2021 20:36:47 +0100
Subject: [PATCH] Process both orientations
Both anticyclonic and cyclonic trajectories.
---
Convert_Matlab/survival.py | 36 +++++++++++++++++++-----------------
1 file changed, 19 insertions(+), 17 deletions(-)
diff --git a/Convert_Matlab/survival.py b/Convert_Matlab/survival.py
index 47abaef7..3bc283af 100755
--- a/Convert_Matlab/survival.py
+++ b/Convert_Matlab/survival.py
@@ -5,24 +5,26 @@ import json
import operator
import csv
-m = sio.loadmat("Trajectories_Anti_filt_v6.mat", squeeze_me=True)
-traj = {}
-belong = []
+for orient1, orient2, orient3 in zip(["cyclo", "anti"], ["Cyclo", "Anti"],
+ ["C", "Antic"]):
+ m = sio.loadmat(f"Trajectories_{orient2}_filt_v6.mat", squeeze_me=True)
+ traj = {}
+ belong = []
-for t in m["Anticyclonic_Trajectories"]:
- assert t.ndim == 1
- assert isinstance(t[0], int)
- traj[t[0]] = t[1].tolist()
-
- for id_det in traj[t[0]]:
- belong.append((id_det, t[0]))
+ for t in m[f"{orient3}yclonic_Trajectories"]:
+ assert t.ndim == 1
+ assert isinstance(t[0], int)
+ traj[t[0]] = t[1].tolist()
-belong.sort(key = operator.itemgetter(0))
+ for id_det in traj[t[0]]:
+ belong.append((id_det, t[0]))
-with open("traj_anti.json", "w") as f:
- json.dump(traj, f, indent = 0)
+ belong.sort(key = operator.itemgetter(0))
-with open("belong.csv", "w", newline='') as f:
- writer = csv.writer(f, lineterminator = "\n")
- writer.writerow(("ID_detection", "ID_traj"))
- writer.writerows(belong)
+ with open(f"traj_{orient1}.json", "w") as f:
+ json.dump(traj, f, indent = 0)
+
+ with open(f"belong_{orient1}.csv", "w", newline='') as f:
+ writer = csv.writer(f, lineterminator = "\n")
+ writer.writerow(("ID_detection", "ID_traj"))
+ writer.writerows(belong)
--
GitLab