51Testing软件测试论坛

标题: 玩转Python之核酸检测日历 [打印本页]
作者: lsekfe    时间: 2022-9-14 13:20
标题: 玩转Python之核酸检测日历
我的坐标是深圳，2022年以来，大部分时候要求24小时，少部分时候要求48小时，更少的时候要求72小时，没有更长的情况。
　　本文根据我的核酸检测记录，制作成日历，将核酸检测记录可视化到日历中。
　　录入数据
　　核酸检测记录能查到的最早时间范围是一个月，以前的检测记录没有提前保存，所以先用8月份的数据制作日历。
　　查询8月份的检测记录，录入到代码中。
  1. # coding=utf-8
  2. 　　from datetime import datetime
  3. 　　# 核酸检测数据，1表示当天做了核酸，0表示当天未做核酸
  4. 　　my_nucleic = {
  5. 　　   'date': [datetime.strftime(datetime(2022, 8, i+1), '%Y-%m-%d') for i in range(31)],
  6. 　　   'nucleic': [1, 1, 0, 1, 0, 1, 0, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1]
  7. 　　}
复制代码
如果当天做了核酸，用1表示，如果当天未做核酸，用0表示。
　　8月的日期使用Python标准库datetime生成。
　　制作日历
　　本文使用Python库openpyxl在excel表格中生成日历。
　　1.使用openpyxl创建表格
import openpyxl
　　# 创建一个workbook对象，而且会在workbook中至少创建一个表worksheet
　　wb = openpyxl.Workbook()
　　# 获取当前活跃的worksheet,默认就是第一个worksheet
　　ws = wb.active

openpyxl是Python中用于读写excel文件的库，pip install openpyxl安装即可使用。
　　2.定义表格初始化和单元格设置的函数
　from openpyxl.styles import PatternFill, Font, Alignment, Border, Side
　　def init_sheet(ws):
　　   for r in range(100):
　　       for c in range(100):
　　           ws.cell(row=r+1, column=c+1).fill = PatternFill('solid', fgColor='000000')
　　def set_cell_style(ws, r, c, color):
　　   ws.cell(row=r, column=c).fill = PatternFill('solid', fgColor=color)
　　   ws.cell(row=r, column=c).font = Font(name="微软雅黑", size=14, bold=True)
　　   ws.cell(row=r, column=c).alignment = Alignment(horizontal='right', vertical='center')
　　   side = Side(style="medium", color="004B3C")
　　   ws.cell(row=r, column=c).border = Border(top=side, bottom=side, left=side, right=side)

　定义一个将表格颜色填充成白色的函数，对表格初始化处理，将背景设置成纯白，日历看起来更美观。
　　定义一个用于处理单元格格式的函数，后面直接调用函数给单元格设置格式，方便重复使用。
　　3.实现日历
import calendar
　　# 将表格填充成白色
　　init_sheet(ws)
　　# 设置年月单元格的边框
　　side = Side(style="medium", color="004B3C")
　　for col in range(7):
　　   ws.cell(row=1, column=col+1).border = Border(top=side, bottom=side, left=side, right=side)
　　# 合并年月单元格
　　ws.merge_cells(start_row=1, start_column=1, end_row=1, end_column=7)
　　# 写入内容和设置格式
　　ws.cell(row=1, column=1).value = '2022年8月'
　　set_cell_style(ws, r=1, c=1, color='418CFA')
　　# 写入星期一至星期日，并设置格式
　　title_data = ['星期一', '星期二', '星期三', '星期四', '星期五', '星期六', '星期日']
　　for col in range(7):
　　   ws.cell(row=2, column=col+1).value = title_data[col]
　　   set_cell_style(ws, r=2, c=col+1, color='418CFA')
　　# 获取一个月的天数和第一天是星期几
　　monthday = calendar.monthrange(2022, 8)
　　# 设置日历的日期
　　col, row = monthday[0], 3
　　for i in range(len(my_nucleic['date'])):
　　   if col < 7:
　　       ws.cell(row=row, column=col + 1).value = i+1
　　       col += 1
　　   else:
　　       col = 0
　　       row += 1
　　       ws.cell(row=row, column=col + 1).value = i+1
　　       col += 1
　　   # 设置单元格格式
　　   set_cell_style(ws, r=row, c=col, color='000000')
　　   # 根据核酸结果填充颜色
　　   if my_nucleic['nucleic'] == 1:
　　       ws.cell(row=row, column=col).fill = PatternFill('solid', fgColor='009B3C')
　　# 设置行高
　　for i in range(1, row+1):
　　   ws.row_dimensions.height = 30
　　# 保存表格
　　wb.save(filename='show_august_nucleic.xlsx')
　　wb.close()

日历效果：
[attach]142858[/attach][attach]142858[/attach]
可以看到，8月份我只有4天没有做核酸，大部分时间都是保持24小时。
　　代码实现介绍：
　　·先合并首行的前7列单元格，写入年月，然后在第二行从左到右写入星期一到星期日，并设置格式。
　　· 使用Python的日历库calendar返回当前月第一天是星期几，然后确定日历的1号的起始位置。
　　· 从1号开始，依次在excel中写入日期，当列到达星期日时，换行并回到星期一的列。
　　· 根据当天是否做了核酸，给单元格填充背景颜色。本文中如果当天做了核酸，则日期的背景设置为绿色（24小时核酸码的颜色），如果未做核酸，则日期的背景设置为白色。
　　· 最后将结果保存到excel文件中，打开excel文件即可看到制作的日历。
　　制作一年的日历
　　制作了一个月的日历后，继续扩展做一年的日历，先看一下效果：
[attach]142859[/attach]
实现方式介绍：
　　· 数据补充，由于只能查到一个月的核酸记录，所以本文除2022年8月外，其他月份的数据用随机数生成。
　　· 对制作一个月日历的代码进行封装，传入年和月，就可以生成任意一个月的日历。
　　· 在excel文件中，自己设计一排展示几个月，本文一排展示3个月。并计算每个月日历的起始单元格位置。
　　· 最后传入年份，依次将一年12个月的日历制作出来，展示在一个页面上。只要有数据，可以对任意一年的日历进行可视化。（代码较长，可在文末获取完整代码）
　　按年的另一种展示方式：
  1. from pyecharts import options as opts
  2. 　　from pyecharts.charts import Calendar
  3. 　　import pandas as pd
  4. 　　nucleic_df = pd.DataFrame()
  5. 　　for i in range(12):
  6. 　　   month_nucleic = made_data(2022, i+1)
  7. 　　   month_df = pd.DataFrame(month_nucleic)
  8. 　　   nucleic_df = pd.concat([nucleic_df, month_df])
  9. 　　data = [[row_data['date'], row_data['nucleic']] for row_index, row_data in nucleic_df.iterrows()]
  10. 　　cal = Calendar(init_opts=opts.InitOpts(width='900px', height='500px'))
  11. 　　cal.add(
  12. 　　   '', data, calendar_opts=opts.CalendarOpts(range_="2022",
  13. 　　                           daylabel_opts=opts.CalendarDayLabelOpts(first_day=1, name_map='cn'))
  14. 　　).set_series_opts(
  15. 　　   label_opts=opts.LabelOpts(font_size=12)
  16. 　　).set_global_opts(
  17. 　　   title_opts=opts.TitleOpts(title='核酸检测日历', pos_left='450', pos_top='0',
  18. 　　       title_textstyle_opts=opts.TextStyleOpts(color='black', font_size=16)),
  19. 　　   visualmap_opts=opts.VisualMapOpts(
  20. 　　       max_=1, min_=0, orient="horizontal", is_piecewise=False,
  21. 　　       range_color=["white", "white", "green"], pos_top="250px", pos_left='50px'
  22. 　　   ),
  23. 　　).render('my_nucleic.html')
复制代码
日历效果：
[attach]142860[/attach]
pyecharts中的Calendar组件也可以实现日历可视化，不过格式比较固定，展示得比较密集。
　　总结
　　·本文用python制作了核酸检测的日历，对核酸检测的天数进行可视化。
　　· 本文用了两种方式实现一年的日历展示。











欢迎光临 51Testing软件测试论坛 (http://bbs.51testing.com/) Powered by Discuz! X3.2