You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
133 lines
4.5 KiB
133 lines
4.5 KiB
import logging
|
|
import pymysql
|
|
import subprocess
|
|
import numpy as np
|
|
from matplotlib import pyplot as plt
|
|
from tkinter import Tk
|
|
from pyecharts import options as _opts
|
|
from pyecharts.charts import Bar
|
|
|
|
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
|
|
|
|
#db_config={'host':'127.0.0.1','user':'root','password':'mysql>hyx123','db':'airquility','charset':'utf8'}
|
|
#ptimelist=[]
|
|
#cilist=[]
|
|
#data_dict=dict(zip(ptimelist,cilist))
|
|
|
|
def get_data(db_config):
|
|
"""
|
|
从数据库中获取ptime和ci数据。
|
|
:param db_config: 数据库配置字典
|
|
:return: 包含ptime和ci的元组列表
|
|
"""
|
|
ptime_list = [] # 存储ptime数据的列表
|
|
ci_list = [] # 存储ci数据的列表
|
|
|
|
|
|
try:
|
|
with pymysql.connect(**db_config) as conn:
|
|
logging.info("数据库连接成功")
|
|
with conn.cursor() as cursor:
|
|
# 使用单个查询以提高性能
|
|
sql_ptime = "SELECT ptime FROM May"
|
|
cursor.execute(sql_ptime)
|
|
data_ptime = cursor.fetchall()
|
|
ptime_list = [row['ptime'] for row in data_ptime if 'ptime' in row]
|
|
#row_list=[val for sublist in row for val in sublist]
|
|
|
|
sql_ci="SELECT ci FROM may"
|
|
cursor.execute(sql_ci)
|
|
data_ci=cursor.fetchall()
|
|
#print(data_ci)
|
|
ci_list = [row['ci'] for row in data_ci if 'ci' in row]
|
|
|
|
sql_so = "SELECT so FROM may"
|
|
cursor.execute(sql_so)
|
|
data_so = cursor.fetchall()
|
|
so_list = [row['so'] for row in data_so if 'so' in row]
|
|
|
|
sql_no = "SELECT no FROM may"
|
|
cursor.execute(sql_no)
|
|
data_no = cursor.fetchall()
|
|
no_list = [row['no'] for row in data_no if 'no' in row]
|
|
|
|
sql_pm = "SELECT pm FROM may"
|
|
cursor.execute(sql_pm)
|
|
data_pm = cursor.fetchall()
|
|
pm_list = [row['pm'] for row in data_pm if 'pm' in row]
|
|
|
|
|
|
return ptime_list, ci_list, so_list, no_list, pm_list
|
|
except pymysql.err.OperationalError as e:
|
|
logging.error(f"数据库操作失败: {e}")
|
|
# 根据具体场景，可以选择返回空列表、抛出异常或返回特定错误码
|
|
return [], [], [], [], []
|
|
#except Exception as e:
|
|
#logging.error(f"未知错误: {e}")
|
|
# 同上，根据具体场景进行处理
|
|
#return [], []
|
|
def visualize_data(ptime_list, ci_list, so_list, no_list, pm_list):
|
|
"""
|
|
可视化数据。
|
|
:param ptime_list: ptime数据列表
|
|
:param ci_list: ci数据列表
|
|
:param column_to_plot: 需要绘制的列名
|
|
"""
|
|
if not ptime_list or not ci_list or not so_list or not no_list or not pm_list:
|
|
logging.error("数据为空，无法进行可视化")
|
|
return
|
|
#y_list=[]
|
|
#for i in range(4,8):
|
|
# y_list.append(i)
|
|
plt.figure(figsize=(15,10),dpi=80)
|
|
x = np.arange(26)
|
|
bar_width = 0.2
|
|
"""
|
|
plt.bar(ptime_list, # 柱状图柱体标签值
|
|
ci_list, # 柱体高度
|
|
align='center', # 柱体对齐方式
|
|
color='red', # 柱体颜色
|
|
tick_label=ptime_list, # 刻度标签值
|
|
hatch='/') # 填充样式，越密集填充就越密
|
|
plt.bar(ptime_list, # 柱状图柱体标签值
|
|
so_list, # 柱体高度
|
|
align='center', # 柱体对齐方式
|
|
color='blue', # 柱体颜色
|
|
tick_label=ptime_list, # 刻度标签值
|
|
hatch='/') # 填充样式，越密集填充就越密
|
|
"""
|
|
plt.xlabel('Ptime')
|
|
plt.bar(ptime_list, ci_list, tick_label=ptime_list, width=bar_width)
|
|
plt.bar(x + bar_width, so_list, width=bar_width)
|
|
#plt.bar(x + bar_width, no_list, width=bar_width)
|
|
#plt.bar(x + bar_width, pm_list, width=bar_width)
|
|
plt.title('CI over time')
|
|
plt.show()
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
db_config = {
|
|
"host": '127.0.0.1',
|
|
"user": "root",
|
|
"password": 'mysql>hyx123',
|
|
"db": 'airquility',
|
|
"charset": 'utf8',
|
|
"cursorclass": pymysql.cursors.DictCursor
|
|
}
|
|
|
|
ptime_list, ci_list, so_list, no_list, pm_list = get_data(db_config)
|
|
visualize_data(ptime_list, ci_list, so_list, no_list, pm_list)
|
|
|
|
print(pymysql.__version__)
|
|
|
|
'''
|
|
window= Tk()
|
|
window.title("每日空气质量")
|
|
window.geometry("300x200")
|
|
#Label(window, text="图表已生成为air_quality_index.html").pack()
|
|
|
|
window.mainloop()
|
|
'''
|
|
|