根据来自一列的不同值获取行

Question

如何在COL2中按不同的值获取行？

例如，我有下面的数据框：

COL1   COL2
a.com  22
b.com  45
c.com  34
e.com  45
f.com  56
g.com  22
h.com  45

我想根据 COL2 中的唯一值获取行：

COL1  COL2
a.com 22
b.com 45
c.com 34
f.com 56

那么，我怎样才能得到呢？如果有人可以提供任何帮助，我将不胜感激。

Answer 1

您可以将Groupby与第一个和最后一个方法结合使用。要从每个组中获取第一行：

df.groupby('COL2', as_index=False).first()

输出：

   COL2   COL1
0    22  a.com
1    34  c.com
2    45  b.com
3    56  f.com

要从每个组中获取最后一行:

df.groupby('COL2', as_index=False).last()

输出：

   COL2   COL1
0    22  g.com
1    34  c.com
2    45  h.com
3    56  f.com

Answer 2

使用drop_duplicates为检查重复项指定列 COL2：

df = df.drop_duplicates('COL2')
#same as
#df = df.drop_duplicates('COL2', keep='first')
print (df)
    COL1  COL2
0  a.com    22
1  b.com    45
2  c.com    34
4  f.com    56

您也可以只保留最后的值：

df = df.drop_duplicates('COL2', keep='last')
print (df)
    COL1  COL2
2  c.com    34
4  f.com    56
5  g.com    22
6  h.com    45

或删除所有重复项：

df = df.drop_duplicates('COL2', keep=False)
print (df)
    COL1  COL2
2  c.com    34
4  f.com    56