Python对Excel求和、合并居中

目录

需求

原始表格:

想在Total列中对每日的Amount进行汇总,然后对Date和Total进行合并居中,效果如下:

思路

遍历Excel行,从第一个非空Date列开始,到下个非空Date列,对Amount列进行求和,结果赋给第一个非空Data列对应行的Total列。

代码

import os
import openpyxl
from openpyxl.styles import Border, Side, PatternFill, Font, GradientFill, Alignment


def range_sum(worksheet,start,end):
    sum = 0
    for row in worksheet[start:end]:
        for cell in row:
            if cell.value != None:
                sum += cell.value
    return sum


def is_blank_row(worksheet,row_num,max_col=None):
    if not max_col:
        max_col = worksheet.max_column 
    for cell in worksheet[row_num][:max_col]:
        if cell.value:
            return False
    return True


def total_amount(worksheet):
    """ 对某sheet的A、E列合并居中,并对E列求和 """
    ws = worksheet
    row, max_row = 4, ws.max_row
    while row < ws.max_row:
        sum_row_start, sum_row_end = row, row
        for working_row in range(row + 1,max_row + 2):
            if (not is_blank_row(worksheet, working_row-1) # 上一行有值
                  and (ws[f'A{working_row}'].value or is_blank_row(worksheet, working_row))): # A列有值 或 当前为空行(最后一次合并)
                    # 求和
                    sum_row_end = working_row - 1
                    ws[f'E{sum_row_start}'] = range_sum(ws,f'C{sum_row_start}',f'C{sum_row_end}')
                    # 合并居中
                    ws[f'E{sum_row_start}'].alignment = Alignment(horizontal="center", vertical="center")
                    ws[f'A{sum_row_start}'].alignment = Alignment(horizontal="center", vertical="center")
                    ws.merge_cells(f'E{sum_row_start}:E{sum_row_end}')
                    ws.merge_cells(f'A{sum_row_start}:A{sum_row_end}')
                    break
        row = sum_row_end + 1


def main():
    # 根据情况修改代码
    in_file_name = 'In.xlsx'
    processing_sheet = 'Sheet1'
    path_name = 'D:\\Users\\Desktop\\Temp'
    out_file_name = 'Out.xlsx'

    wb = openpyxl.load_workbook(filename=os.path.join(path_name,in_file_name))
    total_amount(wb[processing_sheet])
    wb.save(os.path.join(path_name,out_file_name))

if __name__=='__main__':
    main()

说明

  • 本功能用到了openpyxl模块,更多Excel操作请见官网
  • 本代码不支持Excel中间有空行的情况,最后有空行无影响
  • f'A{sum_row_start}'这样的代码用到了f-string功能,若python版本低于3.6,需改为'A'+str(sum_row_start)或其它形式