Keras：更优雅扩展二维NumPy数组维度的方法

你的原始二维数组形状为(7, 6)，需要扩展为Keras要求的(1, 7, 6, 1)形状，以下是几种更简洁的实现方式：

方法1：一次调用np.expand_dims指定多轴 #

不用多次嵌套调用，直接通过axis参数传入多个要插入维度的位置：

input_array_expanded = np.expand_dims(input_array, axis=(0, 0, -1))

axis=(0, 0, -1)表示在数组开头连续插入两个维度，末尾插入一个维度，一步到位得到目标形状。

方法2：用np.reshape直接定义目标形状 #

如果确认原始数组元素总数与目标形状一致（7×6 = 1×7×6×1），可以直接重构形状：

input_array_expanded = input_array.reshape(1, 7, 6, 1)

这种方式最直观，直接指定最终需要的形状，代码简洁易懂。

方法3：利用None（等价于np.newaxis）添加维度 #

通过数组索引中的None占位来快速插入新维度：

input_array_expanded = input_array[None, None, ..., None]

其中...代表保留原数组的所有维度，前后的None分别插入对应位置的新维度，最终得到所需形状。

内容的提问来源于stack exchange，提问作者serotonino