Lagged correlation python