按值对列表求和

S·凯泽(S. Keizer)

我试图通过python中的值求和一个列表,但我没有任何运气。我最后想得出的是三个求和值。

lst = [100, -1, -2, -3, 100, -1, -2, -3, 100]

100 -1-2-3 = 94

100 -1-2-3 = 94

100 = 100
智慧

这个怎么样?

s = []
for i in lst:
    if i > 0 or len(s) == 0:        
        s.append(i)     # start a new element in s when element is larger than zero
    else:
        s[-1] += i      # otherwise add it to the last element of s

s
# [94, 94, 100]

或者,如果您使用熊猫:

import pandas as pd:
ser = pd.Series(lst)
ser.groupby((ser > 0).cumsum()).sum()

#1     94
#2     94
#3    100
#dtype: int64

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章