如何在Pandas中删除列匹配且数值互为相反数的冲销数据行？

我来帮你搞定这个冲销记录清理的问题！你提到用drop_duplicates()没法处理互为相反数的情况，确实如此——这个方法只认完全相同的值。而用groupby的思路是对的，咱们可以按分组列（比如你的a列）来处理每组内的数值，精准识别并删除成对的冲销记录，同时支持奇数个同绝对值异号值的场景（留最后一个剩余值）。

核心思路 #

1. 按分组列（比如a）分组，统计每组内每个数值的绝对值出现次数；
2. 对每个绝对值：
   • 如果出现次数是偶数：说明是成对的冲销记录，全部删除；
   • 如果出现次数是奇数：保留最后一次出现的那个值（你也可以改成保留第一次，只需微调代码）；
3. 筛选出需要保留的行，整理成最终结果。

代码实现 #

首先导入pandas，然后定义处理函数：

import pandas as pd

def remove_reversal_records(df, group_col, value_col):
    # 1. 统计每个分组内，各绝对值的出现次数
    abs_count_df = df.groupby(group_col)[value_col].apply(
        lambda x: x.abs().value_counts()
    ).reset_index()
    
    # 2. 把统计结果转成便于查询的字典：{分组值: {绝对值: 出现次数}}
    count_map = abs_count_df.groupby(group_col).apply(
        lambda x: x.set_index('level_1')[value_col].to_dict()
    ).to_dict()
    
    # 3. 给每行标记绝对值、在组内的累计出现次数
    df['temp_abs'] = df[value_col].abs()
    df['temp_cum_count'] = df.groupby([group_col, 'temp_abs']).cumcount() + 1  # 从1开始计数
    
    # 4. 判断每行是否需要保留：奇数次数留最后一次，偶数次数全删
    df['keep'] = df.apply(
        lambda row: (count_map[row[group_col]][row['temp_abs']] % 2 == 1) 
        and (row['temp_cum_count'] == count_map[row[group_col]][row['temp_abs']]),
        axis=1
    )
    
    # 5. 筛选结果，清理临时列并重置索引
    result = df[df['keep']].drop(['temp_abs', 'temp_cum_count', 'keep'], axis=1).reset_index(drop=True)
    return result

测试你的案例 #

案例1：成对冲销的场景

df1 = pd.DataFrame({"a":["a","b","c","a","a"], "b":[-2,5,2,2,7], "xtra_col":["X","X","X","X","X"]})
result1 = remove_reversal_records(df1, group_col='a', value_col='b')
print(result1)

输出结果：

a  b xtra_col
0  b  5       X
1  c  2       X
2  a  7       X

案例2：奇数个异号值的场景

df2 = pd.DataFrame({"a":["a","b","c","a","a"], "b":[-2,5,2,2.0,-2], "xtra_col":["X","X","X","X","X"]})
result2 = remove_reversal_records(df2, group_col='a', value_col='b')
print(result2)

输出结果：

a    b xtra_col
0  b  5.0       X
1  c  2.0       X
2  a -2.0       X

可选调整 #

如果你想保留第一次出现的剩余值（而不是最后一次），只需要把判断keep的条件改成：

row['temp_cum_count'] == 1

这个方法能灵活处理各种分组下的冲销记录，不管值的顺序如何，只要是同分组内的互为相反数，就能精准识别清理。

内容的提问来源于stack exchange，提问作者callmeGuy