numpy.atleast_3d

numpy.atleast_3d(*arys)

将输入视为至少包含三个维度的数组。

参数：

arys1，arys2，...：array_like

一个或多个阵列样序列。非数组输入转换为数组。已保留三个或更多维度的数组。

返回：

res1，res2，...：ndarray

数组的数组或元组，每个具有a.ndim > = 3。在可能的情况下避免复制，并返回具有三个或更多维度的视图。例如，形状(N,)的1-D数字组变为形状（1， N， 1），并且形状（M， N）的2-D数字组形状（M， N， 1）。

也可以看看

atleast_1d，atleast_2d

例子

>>> np.atleast_3d(3.0)
array([[[ 3.]]])

>>> x = np.arange(3.0)
>>> np.atleast_3d(x).shape
(1, 3, 1)

>>> x = np.arange(12.0).reshape(4,3)
>>> np.atleast_3d(x).shape
(4, 3, 1)
>>> np.atleast_3d(x).base is x
True

>>> for arr in np.atleast_3d([1, 2], [[1, 2]], [[[1, 2]]]):
...     print(arr, arr.shape)
...
[[[1]
  [2]]] (1, 2, 1)
[[[1]
  [2]]] (1, 2, 1)
[[[1 2]]] (1, 1, 2)