From 2c21aab072bca80e59f5ecc8a375dd24ba00edf1 Mon Sep 17 00:00:00 2001
From: Lionel GUEZ <guez@lmd.ens.fr>
Date: Thu, 18 Feb 2021 16:19:10 +0100
Subject: [PATCH] No need for intermediary `var_name`
---
Convert_Matlab/overlap_v6.py | 10 ++--------
1 file changed, 2 insertions(+), 8 deletions(-)
diff --git a/Convert_Matlab/overlap_v6.py b/Convert_Matlab/overlap_v6.py
index ab9ba804..ab4481a8 100755
--- a/Convert_Matlab/overlap_v6.py
+++ b/Convert_Matlab/overlap_v6.py
@@ -41,14 +41,8 @@ def print_edgelist(id_child):
orientation = sys.argv[1] # "cyclo" or "anti"
id_child = sio.loadmat(f"id_child_{orientation}.mat", squeeze_me=True)\
["id_child"]
-
-if orientation == "cyclo":
- var_name = "Ncyclo"
-else:
- var_name = "Nanti"
-
-n_eddies = sio.loadmat(f"{var_name}.mat", squeeze_me = True)[var_name]\
- .astype(int, casting = "safe", copy = False)
+n_eddies = sio.loadmat(f"N{orientation}.mat", squeeze_me = True)\
+ [f"N{orientation}"].astype(int, casting = "safe", copy = False)
date_num = sio.loadmat("date_num.mat", squeeze_me = True)["date_num"]
n_dates = n_eddies.size
print("n_dates =", n_dates)
--
GitLab