以下代码实现了在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]]