Python 数据mapping,缺失日期mapping不上自动跳一行

利用Python做自动化表报时遇到以下情况:

根据需求从数据库中输出数据写到Excel中,并对输出数据进行mapping,如果有缺失数据导致mapping不上的情况,在缺失的那一行自动空行,接着在空行后写入。

如A和B的数据均为mysql 输出数据,如果B缺失日期,则A对应的B为空,在下一行接着写入数据

A B  
data_date total_count data_date total_count data_date_mapping
2018-07-02 4468 2018-07-02 4458 相同
2018-07-09 4561 2018-07-09 4549 相同
2018-07-16 4530 2018-07-16 4517 相同
2018-07-23 4497 2018-07-23 4487 相同
2018-07-30 4416 2018-07-30 4407 相同
2018-08-06 4439 2018-08-06 4432 相同
2018-08-13 4494 2018-08-13 4487 相同
2018-08-27 4603 2018-08-27 4596 相同
2018-09-03 4465 2018-09-03 4457 相同

思路如下:

1.以A的数据为标准,B的数据为变量

2.因元组不能添加元素,因此先把元组转换为list

3.判断,如果B的时间在A中不存在,就插入一个元素,list.insert()

代码如下:

#! /usr/bin/env python
# -*- coding:utf-8 -*-
import MySQLdb
from openpyxl import load_workbook
import sys
import time
import os

reload(sys)
sys.setdefaultencoding('utf8')

db = MySQLdb.connect(host="localhost", user="zimu", passwd="zimu", db="database", port=3306, charset='utf8')
template_file = r"D:\tmepalte_path\template.xlsx"
output_path = r"D:\output_path\\"
if not os.path.exists(output_path):
    os.mkdir(output_path)
book = load_workbook(template_file)
sheet = book['Sheet1']
B_data_list = []

A_sql = '''select '''
B_sql = '''select'''

# 处理A数据
cursor = db.cursor()
cursor.execute(A_sql)
A_data = cursor.fetchall()
t=0
while t<len(A_data):
    for x in range (0,2):
        sheet.cell(t + 3, x + 1).value = A_data[t][x]
    t += 1

# 处理B 数据
cursor.execute(B_sql)
B_data = cursor.fetchall()
B_data_list = list(B_data)

t=0
while t<len(B_data_list):
    if A_data[t][0] != B_data_list[t][0]:
        B_data_list.insert(t,('',''))
    for x in range(0, 2):
        sheet.cell(t + 3, x + 3).value = B_data_list[t][x]
    t += 1

#处理mapping
for i in range(3,sheet.max_row+1):
    if sheet.cell(i,1).value == sheet.cell(i,3).value:
        sheet.cell(i, 5).value = '相同'
    else:
        sheet.cell(i, 5).value = '不同'
book.save(output_path + "tmfs inv mapping_" + time.strftime("%Y%m%d", time.localtime())+ "_" + str(int(time.time())) +".xlsx")


猜你喜欢

转载自blog.csdn.net/u013347671/article/details/83010551