I\'m trying to replace the values of an array with another array of a different shape.
a = np.array([[0,1], [0,0,1]]) b = np.array([[0. 0. 0.],