n******7
发帖数: 12463 | 1 python科学计算当然用numpy这一套
我比较土，要是就算个差额
我还是愿意自己写，这种东西一般也不会是瓶颈
我看了一下diff的实现，
a = asanyarray(a)
nd = len(a.shape)
slice1 = [slice(None)]*nd
slice2 = [slice(None)]*nd
slice1[axis] = slice(1, None)
slice2[axis] = slice(None, -1)
slice1 = tuple(slice1)
slice2 = tuple(slice2)
if n > 1:
return diff(a[slice1]-a[slice2], n-1, axis=axis)
else:
return a[slice1]-a[slice2]
跟zip的思路比较像 |
|
