add value to each element in array python,"[(a + i.reshape(2, 2)) for i in np.identity(4)]"