在 Python 中将第 3 维添加到 2D 数组

Question

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

我怎样才能做到这一点？

Answer 1

使用名为 array 的任何二维形状的数组：

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]