[Numpy-discussion] Easy way to vectorize a loop?