On minor thing that instead of 'ret' there should be 'x'. With kind regards, -gdg
On Dec 12, 2017 22:51, "Joe" <solar...@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@python.org https://mail.python.org/mailman/listinfo/numpy-discussion
_______________________________________________ NumPy-Discussion mailing list NumPy-Discussion@python.org https://mail.python.org/mailman/listinfo/numpy-discussion