Python-熊猫，过滤器取组中的最后一个元素，然后取组中的第一个元素

Question

从这里跟进问题：drop first and last row from within each group

在 pandas 中，如何删除第一个组中的最后一行，然后删除组中所有后续条目的第一行？

例如

      X   Y
a a   0   1
  a   2   3
  c   4   5
  d   6   7
b e   8   9
  f  10  11
  g  12  13
c h  14  15
  i  16  17
d j  18  19

我想要这个

    X   Y
a d 6   7
b e 8   9
c h 14 15
d j 18 19

Answer 1

首先通过get_level_values检查第一级的第一个值，然后通过apply检查groupby - 通过tail检查第一组，通过head检查所有其他值：

first = df.index.get_level_values(0)[0]
df = df.groupby(level=0, sort=False, group_keys=False)
       .apply(lambda x: x.tail(1) if x.name == first else x.head(1))
print (df)
      X   Y
a d   6   7
b e   8   9
c h  14  15
d j  18  19