guilhem_lavaux/vide_public
0
0
Fork 0
mirror of https://bitbucket.org/cosmicvoids/vide_public.git synced 2025-07-04 15:21:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

more general array-getting

Browse source
This commit is contained in:
P.M. Sutter 2014-05-13 03:29:50 -05:00
parent 1aaaeee224
commit c59b393ed3
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
python_tools/void_python_tools/voidUtil/catalogUtil.py
View file

@ -435,10 +435,10 @@ def loadVoidCatalog(sampleDir, dataPortion="central", loadParticles=True):
# -----------------------------------------------------------------------------
def getVoidArray(catalog, attr):
def getArray(objectList, attr):
if hasattr(catalog.voids[0], attr):
return np.fromiter((getattr(v, attr) for v in catalog.voids), float)
if hasattr(objectList[0], attr):
return np.fromiter((getattr(v, attr) for v in objectList), float)
else:
print " Attribute", attr, "not found!"
return -1