def as1D(x):
""" Convert input into to 1D numpy array.
Returns
-------
x : 1D array
Examples
-------
>>> as1D(5)
array([5])
>>> as1D([1,2,3])
array([1, 2, 3])
>>> as1D([[3,4,5,6]])
array([3, 4, 5, 6])
"""
if not isinstance(x, np.ndarray):
x = np.asarray_chkfinite(x)
if x.ndim < 1:
x = np.asarray_chkfinite([x])
elif x.ndim > 1:
x = np.squeeze(x)
return x
评论列表
文章目录