问题描述
我想将 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），以此类推。