numpy矩阵合并

numpy矩阵合并
本文是课程笔记，课程来源：【莫烦Python】Numpy & Pandas (数据处理教程)

• numpy矩阵基本使用
• numpy矩阵计算
• numpy矩阵索引
• numpy矩阵合并
• numpy矩阵分割

import numpy as np

# 此部分为numpy中矩阵合并操作

array_a = np.array([1, 1, 1])
array_b = np.array([2, 2, 2])

# 将a和b按列合并(上下合并)，也就是合并为更高一维度
print(np.vstack((array_a, array_b)))

# 将a和b按行合并(左右合并)，也就是合并为同一维度
print(np.hstack((array_a, array_b)))

# 将一维矩阵a翻转，注意转置无效因为a只有一个维度,于是我们需要用newaxis加一个维度
print(array_a[:, np.newaxis])
# 或者右边加一个维度再转置
print(array_a[np.newaxis, :].T)
# 或者直接reshape
print(array_a.reshape(3, 1))
# 可以直接赋值
array_T = np.array([1, 1, 1])[:, np.newaxis]
print(array_T)

# np.concatenate多矩阵合并,axis决定合并维度
print('axis=0:',
      np.concatenate((array_a[:, np.newaxis], array_b[:, np.newaxis], array_b[:, np.newaxis], array_a[:, np.newaxis]),
                     axis=0))
print('axis=1:',
      np.concatenate((array_a[:, np.newaxis], array_b[:, np.newaxis], array_b[:, np.newaxis], array_a[:, np.newaxis]),
                     axis=1))