import matplotlib.pyplot as plt
import pymysql

plt.rcParams['font.sans-serif'] = ['SimHei']
plt.rcParams['axes.unicode_minus'] = False
# 连接数据库
db = pymysql.connect(host='localhost', user='root', passwd='1234', port=3306, db='epidemic')

# 开启一个游标cursor
cursor = db.cursor()

# 获取phone数据表里的所有数据
sql = 'select cityName,city_confirmedCount from dxyarea where provinceName="山东省" limit 10'

# 执行sql中的语句
cursor.execute(sql)

# 将获取到的sql数据全部显示出来
result = cursor.fetchall()

# 定义需要用上
#
# 的空数据数组,然后通过遍历数据库的数据将数据附上去
xname = []
ynum = []

# 遍历表里的数据
for x in result:
    xname.append(x[0])
    ynum.append(x[1])

# 创建一个figure(一个窗口)来显示条形图
plt.figure()
plt.bar(xname, ynum)
plt.xlabel('cityName')
plt.ylabel('city_suspectedCount')
for x, y in enumerate(ynum):
    plt.text(x, y, '%s' % y)


# 创建一个figure(一个窗口)来显示折线图
plt.figure()
plt.plot(xname, ynum)
for x, y in enumerate(ynum):
    plt.text(x, y, '%s' % y)

# 显示图表
plt.show()

# 关闭游标和数据库
cursor.close()
db.close()