python - 在 Python 中将第三维添加到二维数组

Question

我有一个 2D 虚拟变量数组（0 和 1），形状为 (4432, 35) -> 4432 个视频，包括 35 个不同的客户。由于视频包含 1800 帧，我想用 1800 时间步（帧）向该数组添加第三维，以便它获得形状（4432、35、1800）。所以我希望 Python 将第二维中的零和一乘以 1800 次到第三维。

我怎样才能做到这一点？

标签： pythonarrays

Answer 1

使用具有任何 2D 形状的称为数组的数组：

array = [[[j for k in range(1800)] for j in i] for i in array]

这将创建一个第三维，其中包含第二维中值的 1800 个重复项。

拥有一个形状（4432、1800、35）似乎也更有意义：（视频、框架、框架中的客户）：

array = [[i for k in range(1800)] for i in array]