python - 如何删除两列中具有重复值对的行？

Question

我目前有一个 Pandas DataFrame，想要删除两列中具有重复对的行。这是一个显示我的意思的示例:

    col0    col1    col2
0    0       1       0
1    1       2       1
2    0       3       0
3    4       4       6
4    3       5       1
5    0       6       0

假设我想根据 col0 和 col2 上的值对删除重复项。正如您在此玩具示例中所看到的，行 0、2 和 5 具有重复的 (0, 0) 对，我想删除重复的行并只保留一个。

有没有办法可以有效地做到这一点？我尝试查看 pd.DataFrame.drop_duplicates但它的表现似乎与我的预期略有不同。

提前致谢。

Answer 1

• 您可以传递要删除重复项的列。

import pandas as pd

data = [[0, 1, 0], [1, 2, 1], [0, 3, 0], [4, 4, 6], [3, 5, 1], [0, 6, 0]]

df = pd.DataFrame(data = data, columns=['col0', 'col1', 'col2'])

print(df.drop_duplicates(['col0', 'col2']))

输出:

    col0    col1    col2
0      0       1    0
1      1       2    1
3      4       4    6
4      3       5    1