问在Dataframe中计数连续零的数目

EN

def len_consec_zeros(a):
    a = np.array(list(a))    # convert elements to `str`
    rr = np.argwhere(a == '0').ravel()  # find out positions of `0`
    if not rr.size:  # if there are no zeros, return 0
        return 0

    full = np.arange(rr[0], rr[-1]+1)  # get the range of spread of 0s

    # get the indices where `0` was flipped to something else
    diff = np.setdiff1d(full, rr)
    if not diff.size:     # if there are no bit flips, return the 
        return len(full)  # size of the full range

    # break the array into pieces wherever there's a bit flip
    # and the result is the size of the largest chunk
    pos, difs = full[0], []
    for el in diff:
        difs.append(el - pos)
        pos = el + 1

    difs.append(full[-1]+1 - pos)

    # return size of the largest chunk
    res = max(difs) if max(difs) != 1 else 0

    return res

# join all columns to get a string column

# assuming you have your data in `df`
df['concated'] = df.astype(str).apply(lambda x: ''.join(x), axis=1)
df['consecutive_zeros'] = df.concated.apply(lambda x: len_consec_zeros(x))

ts = pd.Series([0,0,0,0,1,1,0,0,1,1,1,0])
ts2 = 1-ts
tsgroup = ts.cumsum()

consec_0 = ts2.groupby(tsgroup).transform(pd.Series.cumsum)
consec_0.max()

# Inspired by https://stackoverflow.com/a/44385183/
def pos_neg_counts(mask):
    idx = np.flatnonzero(mask[1:] != mask[:-1])
    if len(idx)==0: # To handle all 0s or all 1s cases
        if mask[0]:
            return np.array([mask.size]), np.array([0])
        else:
            return np.array([0]), np.array([mask.size])
    else:
        count = np.r_[ [idx[0]+1], idx[1:] - idx[:-1], [mask.size-1-idx[-1]] ]
        if mask[0]:
            return count[::2], count[1::2] # True, False counts
        else:
            return count[1::2], count[::2] # True, False counts

def get_consecutive_zeros(df):
    arr = df.values
    mask = (arr==0) | (arr=='0')
    zero_count = np.array([pos_neg_counts(i)[0].max() for i in mask])
    zero_count[zero_count<2] = 0
    return zero_count

In [272]: df
Out[272]: 
  DEC JAN FEB  MARCH  APRIL  MAY
0   X   X   X      1      0    1
1   X   X   X      1      0    1
2   0   0   1      0      0    1
3   1   0   0      0      1    1
4   0   0   0      0      0    1
5   X   1   1      0      0    0
6   1   0   0      1      0    0
7   0   0   0      0      1    0

In [273]: df['consecutive_zeros'] = get_consecutive_zeros(df)

In [274]: df
Out[274]: 
  DEC JAN FEB  MARCH  APRIL  MAY  consecutive_zeros
0   X   X   X      1      0    1                  0
1   X   X   X      1      0    1                  0
2   0   0   1      0      0    1                  2
3   1   0   0      0      1    1                  3
4   0   0   0      0      0    1                  5
5   X   1   1      0      0    0                  3
6   1   0   0      1      0    0                  2
7   0   0   0      0      1    0                  4