如何从 3D 制作 2D 数组，使 (100, 50, 20) 为 (100, 100)

【问题标题】：How to make a 2D ndarray from 3D so that (100, 50, 20) is (100, 100)如何从 3D 制作 2D 数组，使 (100, 50, 20) 为 (100, 100)
【发布时间】：2021-02-15 08:15:00
【问题描述】：

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

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

【问题讨论】：

您拥有哪种格式的数据？ numpy ndarray 还是列表？
这个不是很清楚。您想减少数据/维度，但您没有指定如何。没有唯一的解决方案
请添加一段您已经编写过的代码以及示例，包括输入和输出
我已经更新了问题的描述。希望这能让它更清楚。如果您需要更多信息，请告诉我。谢谢。

标签： python arrays numpy numpy-ndarray

【解决方案1】：

只需使用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），依此类推，以此类推。

【讨论】：