numpy中array的合并

import numpy as np

a = np.array([1, 1, 1])
b = np.array([2, 2, 2])

print(np.vstack((a, b)))  # vertical stack 上下合并

print(np.hstack((a, b)))  # horizontal stack 左右合并

print(a[:, np.newaxis])  # 在后面表示在列上增加一个维度
print(a[:, np.newaxis].shape)

print(np.concatenate((a, a), axis=0))  # 指定维度的合并

学习自莫凡Python