Numpy Vectorize Function With Two Arguments