为数组python中的每个元素添加值

Question

我有这样的数组

a= np.arange(4).reshape(2,2)

array([[0, 1],[2, 3]])

我想为数组中的每个元素添加一个值.我希望我的结果返回4个数组

array([[1, 1],[2, 3]])

array([[0, 2],[2, 3]])

array([[0, 1],[3, 3]])

array([[0, 1],[2, 4]])

Answer 1

[a + i.reshape(2, 2) for i in np.identity(4)]