numpy中hstack()和vstack()的简单理解

学习目标：

通过对比一维/二维矩阵进行hstack(horizontal-stack)和vstack(vertical-stack)后的结果，理解hstack和vstack的含义。

代码：

import numpy as np

# 一维矩阵
a = np.array((1, 2, 3))  # 一维数组
b = np.array((4, 5, 6))  # 一维数组

# 把b水平方向(从左往右)推入a
ab_h_stack = np.hstack((a, b))
print(f"ab_h_stack:\n{ab_h_stack}")  # [1 2 3 4 5 6]

# 把a水平推入b
ba_h_stack = np.hstack((b, a))
print(f"ba_h_stack:\n{ba_h_stack}")  # [4 5 6 1 2 3]

# 把b竖直方向(从下往上)推入a
ab_v_stack = np.vstack((a, b))
print(f"ab_v_stack:\n{ab_v_stack}")

# 把a竖直推入b
ba_v_stack = np.vstack((b, a))
print(f"ba_v_stack:\n{ba_v_stack}")


# 二维矩阵
a1 = np.array([[1], [2], [3]])  # 二维数组 (可以简单记忆：1个[ ]就是一维数组，2个[ ]就是二维)
b1 = np.array([[4], [5], [6]])  # 二维数组
print(f"a1:\n{a1}", "\n", f"b1:\n{b1}")

# 把b1水平方向(从左往右)推入a1，得a1b1_h_stack
a1b1_h_stack = np.hstack((a1, b1))
print(f"a1b1_h_st