[Numpy-discussion] What is the pythonic way to write a function that handles arrays and scalars?

[Joe]

Hi,

the best example I found was this one:

https://stackoverflow.com/a/29319864/7919597

def func_for_scalars_or_vectors(x):
     x = np.asarray(x)
     scalar_input = False
     if x.ndim == 0:
         x = x[None]  # Makes x 1D
         scalar_input = True

     # The magic happens here

     if scalar_input:
         return np.squeeze(ret)
     return ret

Is this as good as it gets or do you have other suggestions?

Cheers,
Joe