本文介绍一种简洁、可读性强的方法,利用 `np.dstack` 与 `reshape` 组合,将四个等长一维时间序列快速组装为形状为 `(n, 2, 2)` 的三维数组,每个时间步对应一个 2×2 矩阵。
在时间序列建模、状态空间表示或批量矩阵运算中,常需将多个标量序列(如系统状态变量 A、B、C、D)组织为随时间变化的矩阵序列。例如,对每个时刻 t,构造局部矩阵 [[A[t], B[t]], [C[t], D[t]]],最终得到形状为 (N, 2, 2) 的三维张量。
原始方法通过多次 reshape 和 concatenate 实现,逻辑嵌套深、维度操作易错且难以维护。更优解是借助 np.dstack —— 它沿第三轴(axis=2) 堆叠数组,自动广播为统一形状。对四个 (N,) 向量调用 np.dstack([A, B, C, D]),结果为 (1, N, 4)(注意:dstack 会前置两个单位维度),再经 reshape(N, 2, 2) 即可精准映射为所需结构:
import numpy as np
N = 100
A = np.ones(N) * 1
B = np.ones(N) * 2
C = np.ones(N) * 3
D = np.ones(N) * 4
# 一行代码完成构造
ABCD = np.dstack([A, B, C, D]).reshape(N, 2, 2)
print(ABCD.shape) # (100, 2, 2)
print(ABCD[0]) # [[1. 2.]
# [3. 4.]]
⚠️ 注意事项: