Question

将一个特殊的循环转化为numpy函数

大家好
我在写一个算法，其中需要一个循环，但我怎么想都不知道如何把整个循环转化为numpy函数，可能是对函数库本身还不太熟悉
程序如下：

s = 0
for a in array
s += a
if s < 0:
s = 0

其中累加的运算我找到cumsum函数可以运作，但是if这个部分就完全不知道如何下手。
请高人指点。谢谢！
百度格式太差了，我直接贴个图。

Answer 1

不太明白你这个循环的目的，当s < 0的时候s = 0，这在累加中有什么意义吗？
如果你的这个循环没错，那它已经足够简单了，不需要用numpy来改造，况且cumsum无法对累加的中间值做判断及修改。

追问

你好。
原始数列存在负数的。
算法本身要求累加的时候小于0了必须重置为0。
numpy改造的目的是让计算速度更快。因为哪怕是一个最简单的循环，numpy也比for要快好几个数量级。

经过几天的尝试我已经准备放弃了，你是不是也这么认为？

追答

据我所知没有这样的函数，或者你可以改下cumsum的C源码自己编译。我用reduce试了下性能反而下降。