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

[Kirill Balunov]

On minor thing that instead of 'ret' there should be 'x'.

With kind regards, -gdg

On Dec 12, 2017 22:51, "Joe" <solarjoe at posteo.org> wrote:

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
_______________________________________________
NumPy-Discussion mailing list
NumPy-Discussion at python.org
https://mail.python.org/mailman/listinfo/numpy-discussion
-------------- next part --------------
An HTML attachment was scrubbed...
URL: <http://mail.python.org/pipermail/numpy-discussion/attachments/20171212/72aead27/attachment.html>