numpy.expand_dims¶
-
numpy.
expand_dims
(a, axis)[source]¶ 展开数组的形状。
插入新轴,对应于数组形状中的给定位置。
参数: a:array_like
输入数组。
axis:int
插入新轴的位置(轴之间)。
返回: res:ndarray
输出数组。维数的数量大于输入数组的数量。
例子
>>> x = np.array([1,2]) >>> x.shape (2,)
以下等效于
x[np.newaxis,:]
或x[np.newaxis]
:>>> y = np.expand_dims(x, axis=0) >>> y array([[1, 2]]) >>> y.shape (1, 2)
>>> y = np.expand_dims(x, axis=1) # Equivalent to x[:,newaxis] >>> y array([[1], [2]]) >>> y.shape (2, 1)
请注意,一些示例可以使用
None
而不是np.newaxis
。这些是相同的对象:>>> np.newaxis is None True