python里求累计和

  统计/机器学习 Python    浏览次数:7847        分享
0

比如我一个array或者list,= [2, 1, -1, 3, 5],怎么求它的累加和?

上面那个list的累加和是[2, 3, 2, 5, 10]。


谢谢各位大佬!

 

牛笔小新   2017-12-14 09:07



   2个回答 
3

numpy里有cumsum,就是累加和的意思

>>> import numpy as np
>>> np.cumsum([1, 2, 1])
array([1, 3, 4])


SofaSofa数据科学社区DS面试题库 DS面经

蓝色北方   2017-12-14 09:58

3

补充一个来自python自带的模块itertools的accumulate函数,与numpy的cumsum不同的是它会默认返回一个迭代器。而且当传入一个函数时,可以支持求和以外的其它操作:

from itertools import accumulate
>>> list(accumulate([4,2,3]))
[4, 6, 9]
>>> list(accumulate([4,2,3], max))
[4, 4, 4]



SofaSofa数据科学社区DS面试题库 DS面经

s3040608090   2017-12-14 19:09



  相关讨论

python怎么去除字符串中的连字符?

python怎么只保留某列前几个数字

关于爬虫遇见了"input type=hidden",然后想直接用lxml,不用selenium的办法

dataframe数据变long python有stata类似的命令吗

如何用python获取一个文件的最后修改时间?

如何像访问本地文件系统一样访问云存储?

获取DataFrame所占空间的大小

pyton两个dict如何合并?如果key有重复,取最小值

怎么check dataframe 中的某个元素是否字符串?

怎么对有多个空格的字符串进行split?

  随便看看

z test和t test什么区别?

推荐系统中的召回(recall)是什么意思?

二维numpy.array转为一维的numpy.array或者list

推荐系统里的ALS是什么意思?

为什么梯度的反方向是函数下降最快的方向?