python - Sum slices of consecutive values in a NumPy array -
let's have numpy array a containing 10 values. example situation here, although repeat same array length 100.
a = np.array([1,2,3,4,5,6,7,8,9,10]) i sum first 5 values followed second 5 values , on , store them in new empty list b.
so b contain b = [15,40].
how go doing it?
try list comprehension:
b = [sum(a[current: current+5]) current in xrange(0, len(a), 5)] it takes slices of 5 @ time list, sums them , constructs list. works lists aren't multiple of 5 in length.
(xrange should range in python3+)
Comments
Post a Comment