如何将数组的元素（例如整数）转换为数组，以便在python中获得额外的尺寸

Question

我有一个如下的numpy数组：

<Button theme={this.props.theme} />;

现在我想将其转换为这样的形状：

data=[[1,2,3],[4,5,6]]
data=np.array(data)

这样，我将获得一个额外的维度。如何将数据转换为data2？

Answer 1

使用重塑吗？

data.reshape(2,3,1)
> array([[[1],[2],[3]],
         [[4],[5],[6]]])