Create your own universal function in NumPy