statsmodels.tools.tools.unsqueeze

statsmodels.tools.tools.unsqueeze(data, axis, oldshape)[source]

Unsqueeze a collapsed array.

Parameters:

  • data (ndarray) – The data to unsqueeze.

  • axis (int) – The axis to unsqueeze.

  • oldshape (tuple[int]) – The original shape before the squeeze or reduce operation.

Returns:

The unsqueezed array.

Return type:

ndarray

Examples

>>> from numpy import mean
>>> from numpy.random import standard_normal
>>> x = standard_normal((3,4,5))
>>> m = mean(x, axis=1)
>>> m.shape
(3, 5)
>>> m = unsqueeze(m, 1, x.shape)
>>> m.shape
(3, 1, 5)
>>>