Python数据分析之堆叠数组函数怎么使用

什么是堆叠数组函数

堆叠是一种将几个数组堆叠在一起的操作，可以按照不同的轴进行堆叠。numpy库提供了许多堆叠数组的函数,其中最重要的是vstack()和hstack()函数。vstack()可以堆叠垂直数组（行方向），hstack()可以堆叠水平数组（列方向）。

vstack函数

numpy.vstack(tup)合并两个或多个数组的垂直堆叠。至少有两个数组，如果数组的维数不同，*tup按元素转换为具有相同大小和形状的数组，并沿第一个维度堆叠。

import numpy as np

# 定义两个数组。
array_1 = np.array([[1,2],[3,4]])
array_2 = np.array([[5,6],[7,8]])

# 使用vstack函数垂直堆叠数组。
array_vstacked = np.vstack((array_1, array_2))

print(array_vstacked)

输出如下：

[[1 2]
 [3 4]
 [5 6]
 [7 8]]

hstack函数

numpy.hstack(tup)合并两个或多个数组的水平堆叠。至少有两个数组，如果数组的维数不同，则将它们广播为匹配维度，然后沿第二个维度堆叠。

import numpy as np

# 定义两个数组。
array_1 = np.array([[1,2],[3,4]])
array_2 = np.array([[5,6],[7,8]])

# 使用hstack函数水平堆叠数组。
array_hstacked = np.hstack((array_1, array_2))

print(array_hstacked)

输出如下：

[[1 2 5 6]
 [3 4 7 8]]

总结

堆叠数组函数是numpy库中非常重要的一个功能，可以通过垂直和水平堆叠数组实现不同的操作。同时，这个功能也非常容易使用，只需要调用对应的函数，并传入要堆叠的数组即可。在数据分析中，使用这些函数可以使代码变得更加简单和易于理解。