分配numpy数组

Question

我在创建一个numpy数组时遇到了麻烦。我有一个有一些形状的numpy数组，我正在尝试创建一个具有形状（2，other_array_shape）的新数组。类似的东西：

import numpy as np
x = np.zeros((100, 100))
y = np.zeros((2, i for i in x.shape))

但是，这会返回无效的语法错误。有人能告诉我如何实现这个目标吗？

Answer 1

您只需要连接两个元组：

>>> arr = np.zeros((2,) +  x.shape)
>>> arr.shape
(2, 100, 100)

Answer 2

你需要附加元组：

import numpy as np
x = np.zeros((100, 100))
y = np.zeros((2,) + x.shape)