np.diff二維數組中使用append和prepend,

以下代碼實現了在a的第一列重複一次,再執行diff操作。 這樣就避免了原始diff之後在使用

concatenate

函數進行拼接操作(如果需要的話)。

a=np.array([
    [1,2,3],
    [4,5,6],
    [4, 5, 6],
    [7,8,9],
    [7, 8, 9],
    [1, 8, 4],
])
print(np.diff(a,axis=1,prepend=a[:,0:1]))

 

結果:

 [[ 0  1  1]
 [ 0  1  1]
 [ 0  1  1]
 [ 0  1  1]
 [ 0  1  1]
 [ 0  7 -4]]

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章