我可以在没有循环的情况下清理numpy数组吗？

Question

也许这不是什么大不了的事,但是这让我心碎:

deltas = data[1:] - data[:-1]

有了这个:

for i in range(len(deltas)):
        if deltas[i] < 0: deltas[i] = 0
        if deltas[i] > 100: deltas[i] = 0

对于这个特殊的例子......有更好的方法来做清洁部分吗？

问题第二部分:如果清理规则比这个例子更复杂或更复杂,该怎么办？例如,我们可能只想将所有负数更改为零.或者,我们可能正在进行更复杂的映射.

Answer 1

import numpy as np
deltas=np.diff(data)
deltas[deltas<0]=0
deltas[deltas>100]=0

也可能,而且更快一点

deltas[(deltas<0) | (deltas>100)]=0