Replacing numpy array elements with stacked masks

Question

Consider some array arr and advanced indexing mask mask:

import numpy as np

arr = np.arange(4).reshape(2, 2)
mask = A < 2