Find Index Function In Python