画图
目录:
- 柱状图
- 热力图
- 气泡图
柱状图
import matplotlib.pyplot as plt
# a和b是数字
# 显示中文
plt.rcParams['font.sans-serif'] = ['Microsoft YaHei']
plt.figure(figsize=(6,5))
# 画柱状图
plt.bar(['下雨天','非下雨天'],[a,b],color=['dimgray','royalblue'],width=0.6)
# 加标签
plt.title('白天天气状况',fontsize=20)
plt.ylabel('天数',fontsize=15)
plt.xlabel('天气状况',fontsize=15)
# 柱状图顶端加上文字
plt.text(0,a+2,str(a),fontsize=10)
plt.text(1,b+2,str(b),fontsize=10)
#保存图片
plt.savefig('白天天气状况.png',dpi=300)
热力图
#data1为DataFrame对象
#求协方差矩阵
data_corr = data1.corr()
import seaborn as sns
ax = plt.subplots(figsize=(20, 16))#调整画布大小
# 画热力图
ax = sns.heatmap(data_corr, vmax=.8, square=True, annot=True)
plt.title('热力图',fontsize=30)
plt.savefig('heatmap.png',dpi=200)
气泡图
import numpy as np
# 设置画布
plt.figure(dpi=100,figsize=(10,8))
# a为风力等级的数组[0.0,1.5,2.0,...]
# b为对应风力等级的天数
# alpha = 0.5 设置线条的透明度
# 画气泡
for j in range(7):
plt.scatter(a[j],b[j],color=['red','blue','orange','green','black','gold','gray'][j],
linewidth =0.5,s=b[j], alpha =0.5)
#定义标签
c=[]
for i in a:
c.append(str(i)+'风力等级')
labels=c
#输出数字标签
for i in range(7):
plt.annotate(b[i],xy=(a[i],b[i]),color='b',ha='center', va= 'center')
plt.title('该地区风力等级')
plt.xlabel('风力等级')
plt.ylabel('天数')
# 设置标签位置
plt.legend(labels,labelspacing=1.2)
plt.savefig('该地区风力等级.png',dpi=200)
数据来源为第十届泰迪杯b题