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] 的差異。