Python处理excel合并单元格

读取

可以使用 xlrd，也可以使用 openpyxl，但是 openpyxl 读取不了. xls 格式的 Excel，需要转成 xlsx，有点麻烦，所以使用了 xlrd。

• 打开文件
  xlrd 也遇到了一点问题，文件如果是 xls 文件直接打开，在提取合并单元格的时候会拿不到数据，需要加上 formatting_info=True，formatting_info=True 的时候打开. xlsx 文件会报错 NotImplementedError: formatting_info=True not yet implemented，加个 if 就好了

  workbook = xlrd.open_workbook(path)
  if path.split('.xl')[1] == 's':
  	workbook = xlrd.open_workbook(path, formatting_info=True)
  # 获取sheet
  sheet = workbook.sheet_by_index(0)
  

• 获取所有的合并单元格坐标
  

  # 获取行数
  r_num = sheet.nrows
  # 获取列数
  c_num = sheet.ncols
  merge = sheet.merged_cells
  print(merge)   # [(1, 5, 0, 1), (1, 5, 1, 2)], 对应上面两个合并的单元格
  

• 组装数据

  read_data =[]
  for r in range(r_num):
      li = []
      for c in range(c_num):
      	# 读取每个单元格里的数据，合并单元格只有单元格内的第一行第一列有数据，其余空间都为空
          cell_value = sheet.row_values(r)[c]
          # 判断空数据是否在合并单元格的坐标中，如果在就把数据填充进去
          if cell_value is None or cell_value == '':
              for (rlow, rhigh, clow, chigh) in merge:
                  if rlow <= r < rhigh:
                      if clow <= c < chigh:
                          cell_value = sheet.cell_value(rlow, clow)
          li.append(cell_value)
      read_data.append(li)
  

写入

使用的是 xlwt

• 直接上代码

  f = xlwt.Workbook()
  sheet1 = f.add_sheet('sheet1', cell_overwrite_ok=True)
  headers = ['序号', "姓名", '列表', '1月', '2月', '3月', '4月', '5月']
  name = ['1-AAA', '2-BBB', '3-CCC', '4-DDD']
  status = ['累计应发数', '累计专项扣除-五险一金', '累计免税收入', '累计扣个税']
  # 生成第一行header
  for i in range(0,len(headers)):
      sheet1.write(0,i,headers[i])
  
  # 生成序号、姓名单元格
  i, j = 1, 0
  while i < 10*len(name) and j < len(name):
      sheet1.write_merge(i,i+9,0,0,name[j].split('-')[0])
      sheet1.write_merge(i,i+9,1,1,name[j].split('-')[1])
      i += 10
      j += 1
  
  #生成列表
  i = 0
  while i < 10*len(column0):
      for j in range(0,len(status)):
          sheet1.write(j+i+1,3,status[j])
      i += 10
  # 保存
  f.save(path)
  

Django 导出

• 服务端代码

  excel_stream = io.BytesIO()
  # BytesIO流（在内存中读写）
  f.save(excel_stream)
  res = excel_stream.getvalue()
  excel_stream.close()
  response = HttpResponse(content_type='application/vnd.ms-excel')
  from urllib import parse
  file_name = str(path.name).split('.xls')[0] + '(计算后).xls'
  response['Content-Disposition'] = 'attachment;filename=' + parse.quote(file_name)
  response.write(res)
  return response
  

• 前端
  写一个简单的 form 表单就行了