如何从 3D 制作 2D ndarray，以便 (100, 50, 20) 为 (100, 100)

如何解决如何从 3D 制作 2D ndarray，以便 (100, 50, 20) 为 (100, 100)

我想将 3D 数组 (x,y,z) 的两个维度 (y,z) 合并为一个维度。 y 中的每个对应值都应复制到 z 旁边。

例如。我有 100 帧视频，其坐标为 3 个维度的 15 个关键点。数组形状为 (100,15,3)。我希望输出为 (100,45)，它将 y 和 z 合并为 15x3。

解决方法

只需使用 numpy.reshape。它可用于有选择地展平尺寸。

import numpy as np

mat_3d = np.random.randn(2,3,4)
mat_2d = mat_3d.reshape((mat_3d.shape[0],-1))
print(mat_3d)
print(mat_2d)

在本例中，我使用 (mat_3d.shape[0],-1) 作为 reshape 的参数。这意味着第一个维度必须保持不变，但所有其他维度必须变平（-1 是额外的糖，让 numpy 推断正确的大小，但使用 np.prod(mat_3d.shape[1:]) 将是相同的）。 在这种情况下，Numpy 首先获取最后一个轴（此处为 z）的值，然后是倒数第二个轴（此处为 y），以此类推。