使用大Pandas在一个条件已经达到m后放弃一组中的所有观察结果

Question

我有一个熊猫数据框架，基本上是一个不平衡的面板。我所要做的就是，对于每一组，在条件满足后放弃所有观察。因此，鉴于这个数据集：

person    period    hasTattoo
  A         1          N
  A         2          Y
  A         3          Y
  A         4          Y
  B         1          N
  B         2          N
  C         1          Y
  C         2          Y

我想要这个：

^{pr2}$

我只包括不符合条件的时期，和满足条件的时期，之后什么都不做。在R我可以用数据表. 在

dat[, subCount := 1:.N, by = c("person", "hasTattoo")]
dat <- dat[!(hasTattoo=="Y" & subCount > 1),]

Answer 1

熊猫0.12

def f(df):
    mask = np.cumsum((df.hasTattoo == "Y").values) <= 1
    return df[mask]

df.groupby("person").apply(f)

熊猫0.13

^{pr2}$