Python numpy.trim_zeros函数方法的使用

NumPy（Numerical Python的缩写）是一个开源的Python科学计算库。使用NumPy，就可以很自然地使用数组和矩阵。NumPy包含很多实用的数学函数，涵盖线性代数运算、傅里叶变换和随机数生成等功能。本文主要介绍一下NumPy中trim_zeros方法的使用。

numpy.trim_zeros

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

从一个一维数组或序列中去掉前导and/or后零。

参数：

filt ：1-D array 或 sequence

输入数组。

trim ：str, 可选

一个字符串，其中'f'表示从前部trim，

而'b'表示从后部trim。

默认值为'fb'，从数组的前面和后面trim零。

返回值：

trimmed ：1-D array 或 sequence

trim输入的结果。输入数据类型被保留。

例子

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