以下代碼實現了在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]]