Python中的 numpy.trim_zeros()

在数据处理和统计计算中，经常需要对数据进行清洗和处理。Python中的numpy库提供了众多的函数来方便地进行数据处理和运算。其中，numpy.trim_zeros()函数可以用于去除数据集中开头和结尾的0值。

函数语法和参数

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

• arr: 待处理的数组
• trim: 取值为'fb', 'f', 'b'，分别表示对应在数组开头和结尾进行处理，'f'表示开头，'b'表示结尾。默认值为'fb'，即同时处理数组开头和结尾。

返回值

trim_zeros()函数返回的是去除开头和结尾0值后的新数组。

示例

import numpy as np

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

输出结果为：

[1 2]

如果是只去除结尾的0，则可以通过指定trim参数来控制：

import numpy as np

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

输出结果为：

[1 2]

同理，如果只需要去除开头的0，可以设置trim参数的值为'f'。

注意事项

• trim_zeros()函数只会去除0值，其它数值不会改变。
• 如果数组全是0，那么函数会返回空数组。