<SYSTEM_TASK:>
Solve the following problem using Python, implementing the functions described below, one line at a time
<END_TASK>
<USER_TASK:>
Description:
 def compute(a, b, axis):
 """ Finds optimal displacements localized along an axis """

        delta = []
        for aa, bb in zip(rollaxis(a, axis, 0), rollaxis(b, axis, 0)):
            delta.append(Displacement.compute(aa, bb).delta)
        return LocalDisplacement(delta, axis=axis)