def dist(a, b):
    return sum((i-j)**2 for i, j in zip(a, b))
def test_impl(n):
            df = pd.DataFrame({'A': np.arange(n)})
            df['B'] = df.A.map(lambda a: 2 * a)
            return df.B.sum()