numpy.trim_zeros

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

从一个一维数组或序列中去除前导和/或尾随的零.

参数:

filt1-D 数组或序列

输入数组.

trimstr, 可选

一个带有 ‘f’ 表示从前修剪和 ‘b’ 表示从后修剪的字符串.默认是 ‘fb’,从数组的前后修剪零.

返回:

trimmed1-D 数组或序列

修剪输入的结果.输入数据类型保持不变.

示例

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