数组连接中的vstack()函数与hstack()函数

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,1213]])
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)))

在这种情况下,第一个输出没有问题,可以得到正确输出结果。但是对于后一个输出依旧是有问题,原因是子数组中没有保证行相等的原则。

猜你喜欢

转载自blog.csdn.net/ximu__l/article/details/129367814
今日推荐