From 9d0c6ec49b8cdd826a9759133c6fa4bed6e32d6f Mon Sep 17 00:00:00 2001 From: Jeremy Fee <jmfee@usgs.gov> Date: Wed, 27 Jan 2016 14:34:13 -0700 Subject: [PATCH] Add utility method to list channels in a stream --- geomagio/TimeseriesUtility.py | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) diff --git a/geomagio/TimeseriesUtility.py b/geomagio/TimeseriesUtility.py index 20aa4974..e547e63c 100644 --- a/geomagio/TimeseriesUtility.py +++ b/geomagio/TimeseriesUtility.py @@ -110,3 +110,22 @@ def get_merged_gaps(gaps): if merged_gap is not None: merged_gaps.append(merged_gap) return merged_gaps + + +def get_channels(stream): + """Get a list of channels in a stream. + + Parameters + ---------- + stream : obspy.core.Stream + + Returns + ------- + channels : array_like + """ + channels = {} + for trace in stream: + channel = trace.stats.channel + if channel: + channels[channel] = True + return [ch for ch in channels] -- GitLab