Python numpy ediff1d用法及代码示例

本文简要介绍 python 语言中 numpy.ediff1d 的用法。

用法:

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

数组的连续元素之间的差异。

参数：

ary： array_like

如有必要，将在取差异之前展平。

to_end： 数组，可选

在返回的差异末尾附加的数字。

to_begin： 数组，可选

在返回的差异开头添加的数字。

返回：

ediff1d： ndarray

差异。粗略地说，这是 ary.flat[1:] - ary.flat[:-1] 。

注意：

当应用于掩码数组时，如果使用了 to_begin 和/或 to_end 参数，则此函数会删除掩码信息。

例子：

>>> x = np.array([1, 2, 4, 7, 0])
>>> np.ediff1d(x)
array([ 1,  2,  3, -7])

>>> np.ediff1d(x, to_begin=-99, to_end=np.array([88, 99]))
array([-99,   1,   2, ...,  -7,  88,  99])

返回的数组总是一维的。

>>> y = [[1, 2, 4], [1, 6, 24]]
>>> np.ediff1d(y)
array([ 1,  2, -3,  5, 18])