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

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

1. s = 0
   
2. for a in array
   
3.     s += a
   
4.     if s < 0:
   
5.         s = 0

复制代码

其中累加的运算我找到cumsum函数可以运作，但是if这个部分就完全不知道如何下手。
请高人指点。谢谢！

a = np.array([....])
s = max(0,sum(a)) #s = max(0,np.sum(a))
#改，不好意思，看快了，理解错了

uitb 发表于 2018-4-22 13:33
a = np.array([....])
s = max(0,sum(a)) #s = max(0,np.sum(a))
#改，不好意思，看快了，理解错了{:6_171: ...

非常感谢，这个是不是你最终的回复？
我需要消化理解一下。

uitb 发表于 2018-4-22 13:33
a = np.array([....])
s = max(0,sum(a)) #s = max(0,np.sum(a))
#改，不好意思，看快了，理解错了{:6_171: ...

非常感谢，这个是不是你最终的回复？
我需要消化理解一下。

uitb 发表于 2018-4-22 13:33
a = np.array([....])
s = max(0,sum(a)) #s = max(0,np.sum(a))
#改，不好意思，看快了，理解错了{:6_171: ...

抱歉，你的算法和我的源程序完全不是一回事。
你的算法是累加以后直接和0判断
我的算法是每次累加的时候都要和0判断
不过还是多谢你的回复

bigdeer 发表于 2018-4-23 18:55
非常感谢，这个是不是你最终的回复？
我需要消化理解一下。

这就是原答案，看快理解错了，貌似没法删除回答，就只能补一句了，so sorry !

uitb 发表于 2018-4-23 19:40
这就是原答案，看快理解错了，貌似没法删除回答，就只能补一句了，so sorry ! ...

咦。帖子是可以编辑的呀，可以删除原来的文字的