Python实现Excel负数组合匹配正数并高亮及代码逻辑修正

需求背景 #

Excel的A列包含若干负数（如-1、-3、-6、-8）和一个正数（如11），需用Python找出和的绝对值等于该正数的负数组合，高亮对应单元格并保存输出Excel文件。

原代码的核心问题 #

1. 组合匹配逻辑错误：原代码尝试找负数和等于正数，但实际需求是负数和的绝对值等于正数，即负数的和应为-正数（例如正数是11时，负数和需为-11，绝对值才是11）。
2. 重复值误高亮：若存在重复负数（如两个-3），用cell.value in combination会将所有相同值的单元格高亮，但组合可能仅包含其中一个，导致误标记。
3. 未处理表头/行号对应问题：Pandas读取数据时默认跳过第一行表头，但openpyxl遍历整列会包含表头，数据行号对应逻辑需要调整。

修正后的代码 #

import pandas as pd
import itertools
from openpyxl import load_workbook
from openpyxl.styles import PatternFill

input_file = 'input.xlsx'
output_file = 'output.xlsx'

# 读取Excel，假设第一行是表头；若Excel无表头，添加header=None参数
df = pd.read_excel(input_file)
numbers = df['A'].tolist()

# 分离正数和负数，同时记录每个负数的原始Excel行号
positive_num = None
negative_info = []  # 存储(数值, Excel行号)

for idx, num in enumerate(numbers):
    if isinstance(num, (int, float)):
        if num > 0:
            positive_num = num
        elif num < 0:
            # 数据行从第2行开始（表头占第1行），行号=索引+2；无表头则改为idx+1
            negative_info.append((num, idx + 2))

# 查找所有符合条件的组合对应的行号
def find_combinations(negative_info, target_sum):
    nums = [info[0] for info in negative_info]
    indices = [info[1] for info in negative_info]
    valid_combinations = []
    for r in range(1, len(nums) + 1):
        for subset in itertools.combinations(zip(nums, indices), r):
            subset_nums = [x[0] for x in subset]
            if sum(subset_nums) == target_sum:
                valid_combinations.append([x[1] for x in subset])
    return valid_combinations

# 执行匹配逻辑
valid_row_numbers = []
if positive_num is not None:
    # 修正目标：负数和需等于正数的相反数，绝对值才匹配
    target = -positive_num
    valid_row_numbers = find_combinations(negative_info, target)
else:
    print("未在A列找到正数")

# 高亮对应单元格并保存
wb = load_workbook(input_file)
ws = wb.active
fill = PatternFill(start_color="FFFF00", end_color="FFFF00", fill_type="solid")

for row_list in valid_row_numbers:
    for row in row_list:
        ws[f'A{row}'].fill = fill

wb.save(output_file)
print(f"输出文件已保存为 {output_file}")

关键修改说明 #

• 修正匹配逻辑：将目标和从positive_num改为-positive_num，完全贴合“负数和的绝对值等于正数”的需求。
• 用行号定位单元格：通过记录负数对应的Excel行号，彻底避免重复值导致的误高亮问题。
• 支持多组合匹配：函数返回所有符合条件的组合行号，若只需第一个组合，可在找到后立即return终止循环。
• 明确行号对应关系：根据Excel是否有表头，调整行号计算规则，确保定位准确。

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