欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

Python 对Excel求和、合并居中的操作

程序员文章站 2022-07-06 17:07:16
需求原始表格:想在total列中对每日的amount进行汇总,然后对date和total进行合并居中,效果如下:思路遍历excel行,从第一个非空date列开始,到下个非空date列,对amount列...

需求

原始表格:

Python 对Excel求和、合并居中的操作

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

Python 对Excel求和、合并居中的操作

思路

遍历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)或其它形式

补充:python3 pandas dataframe 对某一列求和

在操作pandas的dataframe的时候,常常会遇到某些列是字符串,某一些列是数值的情况,如果直接使用df_obj.apply(sum)往往会出错

使用如下方式即可对其中某一列进行求和

dataf_test1['diff'].sum() // diff为要求和的列

以上为个人经验,希望能给大家一个参考,也希望大家多多支持。如有错误或未考虑完全的地方,望不吝赐教。