从特定分组的列数据中过滤第一行

Question

数据：

df = pd.DataFrame({
    name':['Jane','Jane','Mike','Mike','Jane','Jane','Jane','Mike','Mike','Jane','Jane','Jane']
    'ctg':['A','P','C','B','B','C','B','E','G','L','M','X'])

预期输出：

name	ctg
Jane	A
Jane	B
Jane	L

我是 python 新手，我想制作新的 Dataframe，其中只包含每个“Jane”名字的第一行。你能请任何人帮助我吗？

Answer 1

您可以使用带有掩码的自定义 groupby：

mask = df['name'].eq('Jane')

df[mask].groupby(df['name'].ne(df['name'].shift()).cumsum(), as_index=False).first()