Python 之 Numpy (五)合并

代码：

import numpy as np

A = np.array([1, 1, 1])  # 序列
B = np.array([2, 2, 2])
# 上下合并
print("上下合并")
C = np.vstack((A, B))
print(A.shape, C.shape)
print(C)
# 左右合并
print("左右合并")
D = np.hstack((A, B))
print(D.shape)
print(D)

运行结果：

上下合并
(3,) (2, 3)
[[1 1 1]
 [2 2 2]]
左右合并
(6,)
[1 1 1 2 2 2]

代码：

# 将横向序列变成纵向序列
print("将横向序列变成纵向序列")
print(A[np.newaxis, :])
print(A[np.newaxis, :].shape)
print(A[:, np.newaxis])

运行结果：

将横向序列变成纵向序列
[[1 1 1]]
(1, 3)
[[1]
 [1]
 [1]]

代码：

# 多个array合并
print("多个array合并")
A = np.array([1, 1, 1])[:, np.newaxis]  # 序列
B = np.array([2, 2, 2])[:, np.newaxis]
print(A)
print("按行合并")
C = np.concatenate((A, B, B, A), axis=0)
print(C)
print("按列合并")
C = np.concatenate((A, B, B, A), axis=1)
print(C)

运行结果：

多个array合并
[[1]
 [1]
 [1]]
按行合并
[[1]
 [1]
 [1]
 [2]
 [2]
 [2]
 [2]
 [2]
 [2]
 [1]
 [1]
 [1]]
按列合并
[[1 2 2 1]
 [1 2 2 1]
 [1 2 2 1]]