Python Vectorized Difficulty