创建示例数据

df = pd.DataFrame({'group': ['A', 'A', 'B', 'B'], 'col1': [1, 2, 3, 4], 'col2': [5, 5, 6, 7]})

按组保留两列值匹配的行，否则按组保留第一行

new_df = df.drop_duplicates(subset=['group'], keep='first')

打印结果

print(new_df)

输出：

group col1 col2 0 A 1 5 2 B 3 6

解释：在上面的示例中，我们使用 drop_duplicates 方法来按组保留两列值匹配的行，否则按组保留第一行。在这里，我们指定了列 'group' 作为 subset 参数。由于 subset 只包含 'group' 一列，因此 pandas 将仅比较各组之间的 'group' 值。最后， keep 参数指定保留第一行。因此，第一个组 A 中的第一行保留，第二个组 B 中的第一行也保留。同时删除了组 A 中的第二行，因为它的 'col2' 值与第一行不匹配，删除了组 B 中的第二行，因为它的 'col1' 和 'col2' 值与第一行匹配。

上一篇：按组保持顺序的最小值和最大值

下一篇：按组保留随机行，使用SAS

按组保留两列值匹配的行，否则按组保留第一行。

创建示例数据

按组保留两列值匹配的行，否则按组保留第一行

打印结果

相关内容

热门资讯