From 8b879c1665ba92a94d31848174dc545d14083f55 Mon Sep 17 00:00:00 2001
From: dave <dave@dtu.dk>
Date: Wed, 8 Jun 2016 08:45:53 +0200
Subject: [PATCH] prepost.windIO.LoadResults: add option to load/save unique
channel name list
---
wetb/prepost/windIO.py | 11 +++++++++++
1 file changed, 11 insertions(+)
diff --git a/wetb/prepost/windIO.py b/wetb/prepost/windIO.py
index 589a8c4e..c7872a2e 100755
--- a/wetb/prepost/windIO.py
+++ b/wetb/prepost/windIO.py
@@ -863,6 +863,17 @@ class LoadResults(ReadHawc2):
return np.array(zvals), np.array(yvals)
+ def save_chan_names(self, fname):
+ """Save unique channel names to text file.
+ """
+ channels = self.ch_df.ch_name.values
+ channels.sort()
+ np.savetxt(fname, channels, fmt='%-100s')
+
+ def load_chan_names(self, fname):
+ dtype = np.dtype('U100')
+ return np.genfromtxt(fname, dtype=dtype, delimiter=';').tolist()
+
def save_csv(self, fname, fmt='%.18e', delimiter=','):
"""
Save to csv and use the unified channel names as columns
--
GitLab