python - 我可以在没有循环的情况下清理 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