Vectorization in Python