fixed bug in getArray

This commit is contained in:
P.M. Sutter 2014-05-28 15:41:29 -05:00
parent cea02c5e31
commit 8b746392b4
2 changed files with 5 additions and 1 deletions

View file

@ -456,9 +456,13 @@ def getArray(objectList, attr):
if hasattr(objectList[0], attr): if hasattr(objectList[0], attr):
ndim = np.shape( np.atleast_1d( getattr(objectList[0], attr) ) )[0] ndim = np.shape( np.atleast_1d( getattr(objectList[0], attr) ) )[0]
attrArr = np.zeros(( len(objectList), ndim )) attrArr = np.zeros(( len(objectList), ndim ))
}
for idim in xrange(ndim): for idim in xrange(ndim):
attrArr[:,idim] = np.fromiter((np.atleast_1d(getattr(v, attr))[idim] \ attrArr[:,idim] = np.fromiter((np.atleast_1d(getattr(v, attr))[idim] \
for v in objectList), float ) for v in objectList), float )
if ndim == 1: attrArr = attArr[:,0]
return attrArr return attrArr
else: else:
print " Attribute", attr, "not found!" print " Attribute", attr, "not found!"

View file

@ -25,7 +25,7 @@ import numpy as np
import os import os
import pylab as plt import pylab as plt
import void_python_tools.apTools as vp import void_python_tools.apTools as vp
from void_python_tools.voidUtil import getArray, shiftPart from void_python_tools.voidUtil import getArray, shiftPart, getVoidPart
def fill_between(x, y1, y2=0, ax=None, **kwargs): def fill_between(x, y1, y2=0, ax=None, **kwargs):
"""Plot filled region between `y1` and `y2`. """Plot filled region between `y1` and `y2`.