分配一個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)