2011年3月30日水曜日

python で list/array の中身をaccumulateする

つまり、
[1, 3, 10, 1, 3]
という list / array から
[1, 4, 14, 15, 18]
を求める。


import numpy
import numpy.random
x = numpy.random.uniform(0, 100, 100)
accum = [sum(x[:i]) for i in range(1, x.shape[0])]


# x が listのとき
x = list(x)
accum = [sum(x[:i]) for i in range(1, len(x))]

0 件のコメント:

コメントを投稿