在相等的列值上切片熊貓數據框

Question

我有一個看起來像這樣的熊貓 df：

import pandas as pd

df = pd.DataFrame({0:[1],5:[1],10:[1],15:[1],20:[0],25:[0],
                   30:[1],35:[1],40:[0],45:[0],50:[0]})
df

列名反映坐標。 我想檢索具有連續相等數字的列的開始和結束坐標。

輸出應該是這樣的：

# start,end
0,15 
20,25
30,35
40,50

Answer 1

IIUCusing groupby與diff和cumsum分裂組

s=df.T.reset_index()
s=s.groupby(s[0].diff().ne(0).cumsum())['index'].agg(['first','last'])
Out[241]: 
   first  last
0             
1      0    15
2     20    25
3     30    35
4     40    50

Answer 2

cumsum來識別 group 和groupby ：

s = df.iloc[0].diff().ne(0).cumsum()
(df.columns.to_series()
 .groupby(s).agg(['min','max'])
)

輸出：

   min  max
0          
1    0   15
2   20   25
3   30   35
4   40   50