From 342efef345e42e2def2a8ffee0834d51a9aa660a Mon Sep 17 00:00:00 2001 From: ignis Date: Mon, 8 Mar 2021 16:24:11 +0900 Subject: [PATCH] add a slot for variable name to save file name format --- binning_v_given_c.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/binning_v_given_c.py b/binning_v_given_c.py index d221e93..8d97ca2 100644 --- a/binning_v_given_c.py +++ b/binning_v_given_c.py @@ -2,6 +2,7 @@ # coding: utf-8 import sys +import os import argparse import numpy as np @@ -19,6 +20,7 @@ params = vars(args) # Parameters xidx = int(params["xindex"]) vfilename = params["variable"] +vname = os.path.splitext(os.path.basename(vfilename))[0] nbins = int(params["nbins"]) boundary = float(params["boundary"]) @@ -104,4 +106,4 @@ arr_dict["cmean"] = np.asarray(cmean) arr_dict["bins"] = np.asarray(bin_edges) arr_dict["count"] = np.asarray(count) -np.savez("cmean_u_given_c_{:03d}".format(xidx), **arr_dict) +np.savez("cmean_{}_given_c_{:03d}".format(vname, xidx), **arr_dict)