mirror of
https://bitbucket.org/cosmicvoids/vide_public.git
synced 2025-07-04 15:21:11 +00:00
fixed bug in getArray
This commit is contained in:
parent
cea02c5e31
commit
8b746392b4
2 changed files with 5 additions and 1 deletions
|
@ -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!"
|
||||||
|
|
|
@ -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`.
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue