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

时间:2010-12-02 19:59:21

标签: python numpy

也许不是什么大不了的事,但是我心碎地跟着这个:

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

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

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

2 个答案:

答案 0 :(得分:9)

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

也可以,而且更快一点

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

答案 1 :(得分:1)

尝试使用numpy.vectorize将函数应用于numpy数组的每个元素。