Python numpy.vstack函数方法的使用

numpy.vstack

numpy.vstack(tup)     [source]

垂直(行)按顺序堆叠数组。

这等效于形状(N,)的1-D数组已重塑为(1,N)后沿第一轴进行concatenation。 重建除以vsplit的数组。

这个函数对于3维以下的数组最有意义。例如，对于具有高度(第一个轴)、宽度(第二个轴)和r/g/b通道(第三个轴)的像素数据。函数concatenate, stack和block提供了更一般的堆叠和连接操作。

例子

1）堆叠一维数组

import numpy as geek 
 
# 输入数组 
in_arr1 = geek.array([1,  2, 3])
in_arr2 = geek.array([4,  5, 6])
 
print("1st Input array:")
print(in_arr1)
print("2nd Input array:")
print(in_arr2)
 
# 垂直堆叠两个数组 
out_arr = geek.vstack((in_arr1,  in_arr2))
print("Output vertically stacked array:")
print(out_arr)

2）堆叠二维数组

import numpy as geek 

# 输入数组 
in_arr1 = geek.array([[1,  2, 3], [-1, -2, -3]])
in_arr2 = geek.array([[4,  5, 6], [-4, -5, -6]])
 
print("1st Input array:")
print(in_arr1)
print("2nd Input array:")
print(in_arr2)
 
# 垂直堆叠两个数组 
out_arr = geek.vstack((in_arr1,  in_arr2))
print("Output stacked array:")
print(out_arr)