Python | Numpy np.ediff1d() 方法

简介

numpy.ediff1d() 函数返回输入数组中相邻元素的差异。生成 n-1 个元素的数组。不同于numpy.diff，ediff1d针对的是数组中每个相邻元素之间的差异，而不是矩阵中相邻的行或列之间的差异。

语法

numpy.ediff1d(arr, to_end = None, to_begin = None)

参数

• arr：输入数组
• to_end：默认值是 None，如果是数组的一个整数，则将该值附加到输出末尾。如果使用布尔值，则此值指示是否在输出之后添加原始数组的值。此参数将在 1.15 版更改为以 prepend 命名。
• to_begin：默认值是 None，如果是数组的整数，则将该值附加到输出开头。如果是 True，则在输出之前添加原始数组的值。

返回值

输出数组的差分。

例子

import numpy as np

a = np.array([1, 2, 4, 7, 0])
print(np.ediff1d(a))

输出：

[ 1  2  3 -7]

b = np.array([[1, 2, 4], [1, 6, 24]])
print(np.ediff1d(b))

输出：

[ 1  2 -3  5 18]

参考链接

• Python | Numpy np.ediff1d() 方法

总结

np.ediff1d() 方法可以很容易地在 numpy 中找到数组中相邻元素的差异。diff 用于相邻的行或列，而 ediff1d 用于相邻元素。在编写机器学习方法时，可以使用此差异将时间序列转换为数据集，或执行任何需要计算相邻值之间的差异以进行数值操作的方法。