有没有更好的方法来查找重复行_including_第一个/最后一个？

Question

考虑一个Pandas数据框：

import pandas as pd

df = pd.DataFrame({
    'a': pd.Series([1,1,1,2,3]),
    'b': pd.Series(list('asdfg'))
})

我想返回列a的重复值的所有行，包括第一行或最后一行。我可以用

做到这一点

df[df['a'].duplicated() | df['a'].duplicated(take_last=True)]

有更好的方法吗？

Answer 1

您可以count出现a并返回重复行values>1。

In [25]: df[(df.groupby('a').transform('count')>1).values]
Out[25]:
   a  b
0  1  a
1  1  s
2  1  d