numpy.trim_zeros

numpy.trim_zeros(filt, trim='fb')[源代码]

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

参数
filt一维数组或序列

输入数组。

trim可选的STR

带“f”的字符串,表示从前面修剪,“b”表示从后面修剪。默认值为“fb”,从数组的前面和后面修剪零。

返回
trimmed一维数组或序列

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

实例

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

输入数据类型保留,list/tuple in表示list/tuple out。

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