Python numpy.trim_zeros函数方法的使用

numpy.trim_zeros 函数用于从一维数组的开头和/或结尾去除零值元素。该函数仅适用于一维数组，如果输入的是多维数组，则会报错。trim_zeros 只去除开头和结尾的零值，而不会影响中间的零值。本文主要介绍一下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输入的结果。输入数据类型被保留。

例子

1）默认去除前后零值

import numpy as np

arr = np.array([0, 0, 1, 2, 0, 3, 0, 0])
result = np.trim_zeros(arr)
print(result)

2）仅去除开头的零值

import numpy as np

arr = np.array([0, 0, 1, 2, 0, 3, 0, 0])
result = np.trim_zeros(arr,'f')
print(result)

3）仅去除结尾的零值

import numpy as np

arr = np.array([0, 0, 1, 2, 0, 3, 0, 0])
result = np.trim_zeros(arr,'b')
print(result)