Counting consecutive numbers in a list

[JasonWilder]

Hi there. I need help assembling a python script that counts the number of consecutive numbers in a Field (Long Int) and writes that count in another field for those numbers. The field comes sorted appropriately.The image is what I am trying to accomplish. I cobbled together an Update Cursor, but having a hard time integrating the counting code into it. Any help is much appreciated.

import arcpy

fc = r'C:\Users\a19573\Desktop\Test\Test.gdb\TestCount'
fields = ['Consecutive','Count']

count_values = {}

with arcpy.da.UpdateCursor (fc, fields) as cursor:
    for row in cursor:
        if not row[0] in count_values.keys():
            count_values[row[0]] = 1
        else:
            count_values[row[0]] += 1
        row[1] = count_values[row[0]]
        cursor.updateRow(row)

### Test Code to integrate into Update Cursor above
random_list=[1,4,5,6,7,9,9,19,21,22,23,24] # Set of consecutive numbers in Field 'Consecutive'
def count_consec(lst):
    consec = [1]
    for x, y in zip(lst, lst[1:]):
        if x == y - 1:
            consec[-1] += 1
        else:
            consec.append(1)
    return consec
print(count_consec(random_list))‍‍‍‍‍‍‍‍‍‍‍‍‍‍‍‍‍‍‍‍‍‍‍‍‍‍‍

[Arne_Gelfert]

So which part of this is not working? Looks like you're trying to tackle this 2 ways.

[JasonWilder]

Hi. Both code blocks work independently of each other. I don't understand how to get my desired result using the return from def count_consec into the update row cursor in order to write the count of consecutive numbers into the Count field. Probably  missing something fundamental. My gut is telling me the count code in the update cursor code block needs to be simplified and rewritten with the code from the def. I appreciate the reply.

[Arne_Gelfert]

Oh, I think, I get it... you don't want to be populating the 'Count' field with the current count, 1 for the 1st, 2 for the 2nd, but with the total count of consecutive numbers. So if there are 7 55's in field 1, field 2 should have the value 7 for all 55's. Is that it?

[JasonWilder]

Yep! The field comes in to me in a specific way, so I don't sort the field or count all instances of specific number. I think about it like making subsets for each range of consecutive numbers the code finds going down the column, then counting how many numbers are in the subset. For example; starting at the first row: 1,2,3,4 (first subset has 4 consecutive), the next number in the sequence is 8 (second subset with only 1 consecutive), next numbers are 34,35,36,37,38 (third subset with 5), next numbers may be 1,2,3,4 again (fourth subset has 4 consecutive), and so on down the rows. Then, the 'Count' field would contain the corresponding count of the subset.

[Arne_Gelfert]

Ok, still different from what I understood But I think I get it now.

[DanPatterson_Retired]

So to clarify, is this what you are looking for?

      [[ 0, 54,  6,  0,  6],   
       [ 1, 34,  3,  6,  9],
       [ 2, 52,  2,  9, 11],
       [ 3, 21,  2, 11, 13],
       [ 4, 55,  1, 13, 14],
       [ 5, 52,  2, 14, 16],
       [ 6, 53,  1, 16, 17],
       [ 7, 55,  1, 17, 18],
       [ 8, 35,  3, 18, 21],
       [ 9, 53,  1, 21, 22]])

for '0'  first sequence starts at 54, is six values long and  begins at index 0 to 6
    '1'  second sequence begins at 34, is 3 long from the 6 to 9 value in the original
    '2'  third sequence 2 long, from the 9 to 11 in the original
    etc‍‍‍‍‍‍‍‍‍‍‍‍‍‍‍‍‍

[JasonWilder]

Hi Dan. Thanks for the reply! I like the numpy approach as it simplifies things quite a bit, but don't think the return is precisely what I need. I should clarify, the random_list was something I was playing with to test the def count_consec code. The def returns counts as I would expect from any list with or without consecutive numbers. Now just figuring out how to leverage all or part of that def or potentially numpy into the Update Cursor code.

[JasonWilder]

Hi Dan. I didn't see your explanation at first in your code block when I was reading it. Let me think about it more. It may be what I need.

[JasonWilder]

Hey Dan. I got study your output a bit more and yes, I think this is what I am after actually. Just need to capture the number in the third column of that array and use it in the cursor so I can write it to the 'Count' field. Any hints on how to do that?