diff --git a/postpic/analyzer/analyzer.py b/postpic/analyzer/analyzer.py index 3204987f..2c67c577 100644 --- a/postpic/analyzer/analyzer.py +++ b/postpic/analyzer/analyzer.py @@ -122,7 +122,7 @@ def identifyspecies(cls, species): ejected_c6b, tracer_proton, protonb """ ret = {'tracer': False, 'ejected': False, 'name': species} - s = species.replace('/', '') + s = species.replace('/', '_') # Regex for parsing ion species name. # See docsting for valid examples diff --git a/postpic/datareader/epochsdf.py b/postpic/datareader/epochsdf.py index 420edca4..ef6a64df 100644 --- a/postpic/datareader/epochsdf.py +++ b/postpic/datareader/epochsdf.py @@ -92,11 +92,12 @@ def grid(self, axis): return self['Grid/Grid/' + axsuffix] def listSpecies(self): - ret = [] + ret = set() for key in self.keys(): - match = re.match('Particles/Px/(\w+)', key) + match = re.match('Particles/\w+/(\w+(/\w+)?)', key) if match: - ret = np.append(ret, match.group(1)) + ret.add(match.group(1)) + ret = list(ret) ret.sort() return ret