Shifting Array In Python