* Added array_tools.array_vstack.
This commit is contained in:
@@ -61,3 +61,27 @@ def array_hstack(arrays):
|
|||||||
return hstack(stk)
|
return hstack(stk)
|
||||||
|
|
||||||
|
|
||||||
|
def array_vstack(arrays):
|
||||||
|
"""Creates a 2D array by vertically stacking many arrays together
|
||||||
|
(along the array's first dimension).
|
||||||
|
Each of the input arrays can be a 1D or 2D array.
|
||||||
|
This function is similar to numpy.vstack.
|
||||||
|
"""
|
||||||
|
from numpy import asarray, vstack
|
||||||
|
stk = []
|
||||||
|
|
||||||
|
for a1 in arrays:
|
||||||
|
a = asarray(a1)
|
||||||
|
dim = len(a.shape)
|
||||||
|
if dim == 1:
|
||||||
|
a = a.reshape((1,len(a)))
|
||||||
|
elif dim == 2:
|
||||||
|
pass
|
||||||
|
else:
|
||||||
|
raise ValueError, "Won't take 3D, 4D, ... arrays"
|
||||||
|
|
||||||
|
stk.append(a)
|
||||||
|
|
||||||
|
return vstack(stk)
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user