numpy.trim_zeros

numpy.trim_zeros(filt, trim='fb')

从一维数组或序列中修剪前导和/或尾随零。

参数：

filt一维数组或序列

输入数组。

trimstr，可选

一个字符串，其中 ‘f’ 表示从前面修剪，‘b’ 表示从后面修剪。默认值为 ‘fb’，即从数组的前后两端修剪零。

返回值：

trimmed一维数组或序列

修剪输入后的结果。输入数据类型将被保留。

示例

>>> import numpy as np
>>> 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])

输入数据类型将被保留，输入列表/元组，输出也为列表/元组。

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