jax.numpy.trim_zerosΒΆ

jax.numpy.trim_zeros(filt, trim='fb')[source]ΒΆ

Trim the leading and/or trailing zeros from a 1-D array or sequence.

LAX-backend implementation of trim_zeros(). Original docstring below.

Parameters
  • filt (1-D array or sequence) – Input array.

  • trim (str, optional) – A string with β€˜f’ representing trim from front and β€˜b’ to trim from back. Default is β€˜fb’, trim zeros from both front and back of the array.

Returns

trimmed – The result of trimming the input. The input data type is preserved.

Return type

1-D array or sequence

Examples

>>> a = np.array((0, 0, 0, 1, 2, 3, 0, 2, 1, 0))
>>> np.trim_zeros(a)
array([1, 2, 3, 0, 2, 1])
>>> np.trim_zeros(a, 'b')
array([0, 0, 0, ..., 0, 2, 1])

The input data type is preserved, list/tuple in means list/tuple out.

>>> np.trim_zeros([0, 1, 2, 0])
[1, 2]