vstack()与hstack()用法

import numpy as np

a = np.array([1, 2])
b = np.array([3, 4])
c = [5, 6]

ans = np.vstack((a, b, c))
print("一维数组垂直堆叠\n",ans)

ans = np.hstack((a, b, c))
print("一维数组水平堆叠\n",ans)


a = np.array([[1, 2],[3, 4]])
b = np.array([[5, 6],[7, 8]])

ans = np.vstack((a, b))
print("二维数组垂直堆叠\n",ans)

ans = np.hstack((a, b))
print("二维数组水平堆叠\n",ans)

一维数组垂直堆叠
 [[1 2]
 [3 4]
 [5 6]]
一维数组水平堆叠
 [1 2 3 4 5 6]
二维数组垂直堆叠
 [[1 2]
 [3 4]
 [5 6]
 [7 8]]
二维数组水平堆叠
 [[1 2 5 6]
 [3 4 7 8]]