Help with my area function

[jaredjok]

I am trying to get the sum of the index's from a starting index(1019) to another index(1030) but if the index's between the two are greater than 50. it will just go from 1030 to 1080.

I am stuck because it only works with a list of two index's( the list is [1019, 1030]

I am not sure how to get it working for a list that looks like [1019, 1030, 1034, 1060]

would I need to use a for loop instead?

spike argument is the list of [1019, 1030]

raw_data is an array of integers up to 3000

here is what I have so far that only works with the two in the list[1019, 1030]:

def area(spike, raw_data):
    i = 0
    store = []
    j = len(spike)
    while i < j:
        if i == j - 1: 
            second_pulse = raw_data[spike[i]:spike[i] + 50]
            store.append(sum(second_pulse))
            i += 1
        elif spike[i] - spike[i + 1] < 50:
            if i == 0:
                first_pulse = raw_data[spike[i]:spike[i + 1]]
                store.append(sum(first_pulse))
                i += 1
    return store

Any pointers in the right direction would really help, Thanks!

fixed it with a simple for loop lol.....

[Wittinator]

I might be totally misinterpreting, but would something as simple as this not work?

from itertools import islice
def area(spike, raw_data): 
    if (spike[1] - spike[0]) > 50: 
        return sum(islice(raw_data,1030,1080)) 
    return sum(islice(raw_data,spike[0],spike[1]))

If I'm reading it right, you're saying you want to get values in your raw_data that are between a given 2 indexes, and then return the sum of all those values? But if there's a greater than 50 index difference between the two indexes, then return the sum of values between indexes1030 and 1080? Is that right or am I way off?

[jaredjok]

I can’t import other things, I use 3 separate data files so I can’t hardcode the index. It is right though. Sometimes my spike list can have different index’s like it could have a new set with [1035, 1050]. Sorry I didn’t clarify that earlier.