numpy 扩充一个维度

你可以使用 numpy 的 reshape() 方法来扩充一个维度。假设你有一个形状为 (3, 4) 的数组 arr，你可以使用 reshape() 方法将其转换为一个形状为 (3, 4, 1) 的三维数组，如下所示：

import numpy as np

arr = np.array([[1, 2, 3, 4],
                [5, 6, 7, 8],
                [9, 10, 11, 12]])

new_arr = arr.reshape(3, 4, 1)

在这个例子中，reshape() 方法将 arr 转换为一个形状为 (3, 4, 1) 的三维数组，其中最后一个维度的大小为 1。这个新的维度可以用来存储额外的信息，例如颜色通道或时间步等。注意，reshape() 方法将返回一个新的数组，而不会修改原始数组。