jan/meanas
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

unvec to (3, *shape) rather than list-of-ndarrays

Browse Source
This commit is contained in:
Jan Petykiewicz 2019-07-09 20:12:48 -07:00
parent 2acbda4764
commit d462ae9412
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
fdfd_tools/vectorization.py
View File

@ -45,5 +45,5 @@ def unvec(v: vfield_t, shape: numpy.ndarray) -> field_t:
""" """
if numpy.any(numpy.equal(v, None)): if numpy.any(numpy.equal(v, None)):
return None return None
return [vi.reshape(shape, order='C') for vi in numpy.split(v, 3)] return vi.reshape((3, *shape), order='C')