vstack()函数与hatack()函数
vstack()函数能够在垂直方向上将两个数组连接起来。
hatack()函数能够在水平方向上将两个数组连接起来。
数组类型为一维数组时
下面看代码显示:
import numpy as np
arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])
# 竖直方向上连接
print(np.vstack((arr1, arr2)))
#水平方向上连接
print(np.hstack((arr1, arr2)))
输出的结果为:
[[1 2 3]
[4 5 6]]
[1 2 3 4 5 6]
数组类型为二维数组时
代码显示:
import numpy as np
arr1 = np.array([[1, 2], [3, 4], [5, 6]])
arr2 = np.array([[7, 8], [9, 10], [11,12]])
print(np.vstack((arr1, arr2)))
print(np.hstack((arr1, arr2)))
输出的结果为:
[[ 1 2]
[ 3 4]
[ 5 6]
[ 7 8]
[ 9 10]
[11 12]]
[[ 1 2 7 8]
[ 3 4 9 10]
[ 5 6 11 12]]
两个相连接的数组类型大小不一致时
如果代码如下所示则会出错:
情况一
import numpy as np
arr1 = np.array([[1, 2], [3, 4], [5, 6]])
arr2 = np.array([[7, 8], [9, 10], [11]])
print(np.vstack((arr1, arr2)))
print(np.hstack((arr1, arr2)))
# 此种情况输出错误
情况二
import numpy as np
arr1 = np.array([[1, 2], [3, 4], [5, 6]])
arr2 = np.array([[7, 8], [9, 10], [11,12,13]])
print(np.vstack((arr1, arr2)))
print(np.hstack((arr1, arr2)))
# 此种情况输出错误
情况三:
import numpy as np
arr1 = np.array([[1, 2], [3, 4], [5, 6]])
arr2 = np.array([[7, 8], [9, 10]])
print(np.vstack((arr1, arr2)))
print(np.hstack((arr1, arr2)))
在这种情况下,第一个输出没有问题,可以得到正确输出结果。但是对于后一个输出依旧是有问题,原因是子数组中没有保证行相等的原则。