Commit 4ac62423 authored by Daniel Brown's avatar Daniel Brown
Browse files

fixing output selector

parent 50d0ed53
...@@ -75,14 +75,14 @@ class katRun(object): ...@@ -75,14 +75,14 @@ class katRun(object):
def __getitem__(self, value): def __getitem__(self, value):
idx = [i for i in range(len(self.ylabels)) if self.ylabels[i].split(" ")[0] == str(value)] idx = [i for i in range(len(self.ylabels)) if self.ylabels[i].split(" ")[0] == str(value)]
if len(idx) > 0 and self.y.shape == (): if len(idx) > 0 and self.y.shape == ():
# In the case we have a noxaxis and just one output... # In the case we have a noxaxis and just one output...
return float(self.y) return float(self.y)
elif len(idx) == 1 and len(self.y.shape) >= 1: elif len(idx) == 1 and len(self.y.shape) == 1:
# only selecting a single output from a 1D array # only selecting a single output from a 1D array
if self.y.size == 1: if self.y.size == 1:
return self.y return self.y
else: else:
return self.y[idx[0]] return self.y[idx[0]]
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment