dask.array.squeeze

dask.array.squeeze¶

dask.array.squeeze(a, axis=None)[源代码]¶

从 a 中移除长度为一的轴。

此文档字符串是从 numpy.squeeze 复制的。

Dask 版本可能存在一些不一致性。

参数

aarray_like: 输入数据。
轴None 或 int 或 int 的元组，可选: 1.7.0 新版功能.

选择形状中长度为一的条目的子集。如果选择形状条目大于一的轴，则会引发错误。

返回

压缩的ndarray: 输入数组，但移除了长度为1的所有或部分维度。这总是 a 本身或 a 的一个视图。注意，如果所有轴都被压缩，结果是一个0维数组而不是标量。

Raises

ValueError: 如果 axis 不是 None，并且被挤压的轴的长度不是 1

参见

expand_dims: 逆操作，添加长度为一的条目
reshape: 插入、移除和合并维度，并调整现有维度的大小

示例

>>> import numpy as np  
>>> x = np.array([[[0], [1], [2]]])  
>>> x.shape  
(1, 3, 1)
>>> np.squeeze(x).shape  
(3,)
>>> np.squeeze(x, axis=0).shape  
(3, 1)
>>> np.squeeze(x, axis=1).shape  
Traceback (most recent call last):
...
ValueError: cannot select an axis to squeeze out which has size
not equal to one
>>> np.squeeze(x, axis=2).shape  
(1, 3)
>>> x = np.array([[1234]])  
>>> x.shape  
(1, 1)
>>> np.squeeze(x)  
array(1234)  # 0d array
>>> np.squeeze(x).shape  
()
>>> np.squeeze(x)[()]  
1234

dask.array.square

dask.array.stack