Python NumPy diff方法用法及代码示例

例子

基本用法

a = np.array([1, 3, 8, 15, 30])
np.diff(a)



array([ 2,  5,  7, 15])

递归计算差异

假设我们要递归计算两次差异，即 n=2 。 diff(~) 首先计算 n=1 时的情况，然后对其输出执行另一个 diff(~)。

n=1 时的情况：

a = np.array([1, 3, 8, 15, 30])
np.diff(a, n=1)



array([ 2, 5, 7, 15])

n=2 时的情况：

a = np.array([1, 3, 8, 15, 30])
np.diff(a, n=2)



array([3, 2, 8])

观察 n=2 只是在 n=1 的输出上应用 diff 方法。

计算二维数组的差异

考虑以下二维数组

a = np.array([[1, 3], [8, 15]])
a



array([[ 1,  3],
       [ 8, 15]])

逐行

np.diff(a)   # or axis=1



array([[2],
       [7]])

按列

np.diff(a, axis=0)



array([[ 7, 12]])

在计算之前预置值

a = np.array([3, 8, 15, 30])
np.diff(a, prepend=1)



array([ 2,  5,  7, 15])

在这里，我们将值 1 添加到 a 前面，因此本质上我们正在计算 [1,3,8,15,30] 的差异。