numpy.vstack 是 NumPy 库中的一个函数,用于垂直堆叠输入数组的序列以形成单个数组。这个函数在数据处理和科学计算中非常有用,特别是在需要将多个数组沿着垂直方向(行方向)合并的时候。本文主要介绍一下NumPy中vstack方法的使用。

numpy.vstack

numpy.vstack(tup)     [source]

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

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

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

参数 :

tup :ndarrays的sequence

除第一轴外,所有阵列的形状都必须相同。 

一维数组必须具有相同的长度。

返回值 :

stackedndarray

通过堆叠给定阵列形成的阵列将至少为二维。

例子

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)

推荐文档

相关文档

大家感兴趣的内容

随机列表