Rolling Function In Python With Parameters