python - 从每个组中删除第一行和最后一行

Question

这是对 get first and last values in a groupby 的后续问题

如何删除每个组中的第一行和最后一行？

我有这个df

df = pd.DataFrame(np.arange(20).reshape(10, -1),
                  [['a', 'a', 'a', 'a', 'b', 'b', 'b', 'c', 'c', 'd'],
                   ['a', 'a', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j']],
                  ['X', 'Y'])

df

我故意让第二行与第一行具有相同的索引值。我无法控制索引的唯一性。

      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 b   2.0   3
  c   4.0   5
b f  10.0  11

因为 0 级等于 'c' 和 'd' 的两个组都少于 3 行，所以应该删除所有行。

Answer 1

我将对另一个问题应用类似的技术:

def first_last(df):
    return df.ix[1:-1]

df.groupby(level=0, group_keys=False).apply(first_last)