CODEWITHSUNDEEP
pythonarrays
Charles Hay
Charles Hay

Reputation: 3

Python - how to add and subtract elements in array

If I have an array, let's say: np.array([4,8,-2,9,6,0,3,-6]) and I would like to add the previous number to the next element, how do I do? And every time the number 0 shows up the addition of elements 'restarts'.

An example with the above array, I should get the following output when I run the function:

stock = np.array([4,12,10,19,25,0,3,-3]) is the right output, if the above array is inserted in transactions.

def cumulativeStock(transactions):

    # insert your code here

    return stock

I can't think of a method to solving this problem. Any help would be very appreciated.

Upvotes: 0

Views: 3084

Answers (5)

suvy
suvy

Reputation: 693

def cumulativeStock(transactions):

    def accum(x):
        acc=0
        for i in x: 
           if i==0:
              acc=0
           acc+=i
           yield acc

    stock = np.array(list(accum(transactions)))
    return stock

for your input np.array([4,8,-2,9,6,0,3,-6]) it returns array([ 1, 3, 6, 9, 13, 0, 1, 3, 6])

Upvotes: 0

Sayali Sonawane
Sayali Sonawane

Reputation: 12599

import numpy as np
stock = np.array([4, 12, 10, 19, 25,  0,  3, -3, 4, 12, 10, 0, 19, 25,  0,  3, -3])

def cumsum_stock(stock):
    ## Detect all Zero's first 
    zero_p = np.where(stock==0)[0]
    ## Create empty array to append final result  
    final_stock = np.empty(shape=[0, len(zero_p)])
    for i in range(len(zero_p)):
        ## First Zero detection
        if(i==0):
             stock_first_part = np.cumsum(stock[:zero_p[0]])
             stock_after_zero_part  = np.cumsum(stock[zero_p[0]:zero_p[i+1]])
             final_stock = np.append(final_stock, stock_first_part)
             final_stock = np.append(final_stock, stock_after_zero_part) 
        ## Last Zero detection
        elif(i==(len(zero_p)-1)):  
             stock_last_part = np.cumsum(stock[zero_p[i]:])
             final_stock = np.append(final_stock, stock_last_part, axis=0)
        ## Intermediate Zero detection
        else:
            intermediate_stock = np.cumsum(stock[zero_p[i]:zero_p[i+1]])
            final_stock = np.append(final_stock, intermediate_stock, axis=0)
    return(final_stock)


final_stock = cumsum_stock(stock).astype(int)

#Output
final_stock
Out[]: array([ 4, 16, 26, ...,  0,  3,  0])

final_stock.tolist()
Out[]: [4, 16, 26, 45, 70, 0, 3, 0, 4, 16, 26, 0, 19, 44, 0, 3, 0]

Upvotes: 0

Nils Werner
Nils Werner

Reputation: 36739

another vectorized solution:

import numpy as np
stock = np.array([4, 8, -2, 9, 6, 0, 3, -6])

breaks = stock == 0
tmp = np.cumsum(stock)
brval = numpy.diff(numpy.concatenate(([0], -tmp[breaks])))
stock[breaks] = brval
np.cumsum(stock)
# array([ 4, 12, 10, 19, 25,  0,  3, -3])

Upvotes: 0

supernl
supernl

Reputation: 71

I assume you mean you want to seperate the list at every zero? 

from itertools import groupby
import numpy


def cumulativeStock(transactions):

#split list on item 0
groupby(transactions, lambda x: x == 0)
all_lists = [list(group) for k, group in groupby(transactions, lambda x: x ==  0) if not k]

# cumulative the items
stock = []
for sep_list in all_lists:
    for item in numpy.cumsum(sep_list):
        stock.append(item)

return stock


print(cumulativeStock([4,8,-2,9,6,0,3,-6]))

Which will return: [4, 12, 10, 19, 25, 3, -3]

Upvotes: -1

zwep
zwep

Reputation: 1340

I believe you mean something like this?

z = np.array([4,8,-2,9,6,0,3,-6])
n = z == 0
    [False False False False False  True False False]
res = np.split(z,np.where(n))
    [array([ 4,  8, -2,  9,  6]), array([ 0,  3, -6])] 
res_total = [np.cumsum(x) for x in res]
    [array([ 4, 12, 10, 19, 25]), array([ 0,  3, -3])]
np.concatenate(res_total)
    [ 4 12 10 19 25  0  3 -3]

Upvotes: 2

Related Questions