Given two numpy arrays of equal shape, I want to track how elements from the first array have moved in