如何通过索引将数组插入现有的多维数组？ [蟒蛇]

Question

我通过向数据集中的每个数字添加一组数字来操纵MNIST数据集进行研究。

操作之前：

In:
x_train.shape

Out: 
(60000, 28, 28)

处理后的预期结果：

In:
x_train_new.shape

Out: 
(60000, 11, 28, 28)

但是我搞砸了，忘了在每次迭代中添加x_train[i]。因此，我的形状如下：

In:
x_train_new.shape

Out: 
(60000, 10, 28, 28)

我尝试应用np.insert，但是由于它是一个多维数组，我在努力正确地应用它：

tryout = x_train_new[0]

In:
np.insert(tryout, 0, x_train[0])

Out:
ValueError: could not broadcast input array from shape (28,28) into shape (28)

如何在每个``ì`''中插入x_train[i]？是否可以将数组作为每个数组的第一个值插入？

感谢您的帮助。非常感谢。

Answer 1

只需创建一个空数组

x_train_expected = np.empty((60000, 11, 28, 28))

然后做

x_train_expected[:,1:] = x_train_new
x_train_expected[:,0] = x_train