numpy.trim_zeros() in Python (original) (raw)
Last Updated : 27 Sep, 2019
numpy.trim_zeros function is used to trim the leading and/or trailing zeros from a 1-D array or sequence.
Syntax: numpy.trim_zeros(arr, trim)
Parameters:
arr : 1-D array or sequence
trim : trim is an optional parameter with default value to be ‘fb'(front and back) we can either select ‘f'(front) and ‘b’ for back.Returns: trimmed : 1-D array or sequence (without leading and/or trailing zeros as per user’s choice)
Code 1:
import
numpy as geek
gfg
=
geek.array((
0
,
0
,
0
,
0
,
1
,
5
,
7
,
0
,
6
,
2
,
9
,
0
,
10
,
0
,
0
))
res
=
geek.trim_zeros(gfg)
print
(res)
**Output :**array([1, 5, 7, 0, 6, 2, 9, 0, 10])
Code 2:
import
numpy as geek
gfg
=
geek.array((
0
,
0
,
0
,
0
,
1
,
5
,
7
,
0
,
6
,
2
,
9
,
0
,
10
,
0
,
0
))
res
=
geek.trim_zeros(gfg,
'f'
)
print
(res)
**Output :**array([1, 5, 7, 0, 6, 2, 9, 0, 10, 0, 0])
Code 3:
import
numpy as geek
gfg
=
geek.array((
0
,
0
,
0
,
0
,
1
,
5
,
7
,
0
,
6
,
2
,
9
,
0
,
10
,
0
,
0
))
res
=
geek.trim_zeros(gfg,
'b'
)
print
(res)
**Output :**array([0, 0, 0, 0, 1, 5, 7, 0, 6, 2, 9, 0, 10])