From c59b393ed351ee8ea801275a2f2d99ab07ab2a60 Mon Sep 17 00:00:00 2001 From: "P.M. Sutter" Date: Tue, 13 May 2014 03:29:50 -0500 Subject: [PATCH] more general array-getting --- python_tools/void_python_tools/voidUtil/catalogUtil.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/python_tools/void_python_tools/voidUtil/catalogUtil.py b/python_tools/void_python_tools/voidUtil/catalogUtil.py index 9e2bdb4..d3f007c 100644 --- a/python_tools/void_python_tools/voidUtil/catalogUtil.py +++ b/python_tools/void_python_tools/voidUtil/catalogUtil.py @@ -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