Numpy vectorize as a decorator with arguments Numpy vectorize as a decorator with arguments numpy numpy