python - pandas groupby 在一个分类系列中运行

Question

给定一个系列:

import numpy as np
import pandas as pd

N = 18
color = np.repeat(list("ABCABC"), 3)
ss = pd.Series(color, index=pd.date_range("2020-01-01", periods=N,
                                          freq="10s", tz="UTC"),
               dtype="category")

>>> print(ss)
2020-01-01 00:00:00+00:00    A
2020-01-01 00:00:10+00:00    A
2020-01-01 00:00:20+00:00    A
2020-01-01 00:00:30+00:00    B
2020-01-01 00:00:40+00:00    B
2020-01-01 00:00:50+00:00    B
2020-01-01 00:01:00+00:00    C
2020-01-01 00:01:10+00:00    C
2020-01-01 00:01:20+00:00    C
2020-01-01 00:01:30+00:00    A
2020-01-01 00:01:40+00:00    A
2020-01-01 00:01:50+00:00    A
2020-01-01 00:02:00+00:00    B
2020-01-01 00:02:10+00:00    B
2020-01-01 00:02:20+00:00    B
2020-01-01 00:02:30+00:00    C
2020-01-01 00:02:40+00:00    C
2020-01-01 00:02:50+00:00    C
Freq: 10S, dtype: category
Categories (3, object): [A, B, C]

我怎样才能对它进行分组，以便类别的连续级别的每个块都是它自己的组？在示例中，这意味着最终有 6 个组。

Answer 1

使用 shift与 cumsum创建子键

m = ss.ne(ss.shift()).cumsum()
ss.groupby(m).ngroups
6