statsmodels.tools.tools.unsqueeze

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

Unsqueeze a collapsed array.

Parameters:
datandarray

The data to unsqueeze.

axisint

The axis to unsqueeze.

oldshapetuple[int]

The original shape before the squeeze or reduce operation.

Returns:
ndarray

The unsqueezed array.

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)
>>>

Last update: Nov 14, 2024