def get_spike_indices()

in crashclouseau/utils.py [0:0]

• 4 lines of code
• 5 McCabe index (conditional complexity)

def get_spike_indices(numbers, ndays):
    """Get the spikes indices from the numbers (list)"""
    # we've something like [0, 0, 0, 2, 0, 0, 0, 3, 1, 0, 0, 9] and ndays=3
    # and we want to get [3, 7]
    for i in range(ndays, len(numbers)):
        if compare_numbers(numbers[i], numbers[(i - ndays):i]):
            yield i