一、基础设置
导入相关的库
import pandas as pd import numpy as np import matplotlib.pyplot as plt import seaborn as sns %matplotlib inline #在ipython总显示图表
默认不显示中文,因此需要更改设置,显示中文
#显示中文 import matplotlib as mpl mpl.rcParams[‘font.sans-serif‘] = [u‘SimHei‘] mpl.rcParams[‘axes.unicode_minus‘] = False
设置全局变量
mpl.rc(‘font‘,size=12) #字体 mpl.rc(‘figure‘,figsize=(8,6)) #图像大小 mpl.rc(‘axes.spines‘,right=False,top=False) #设置右边上边的横线是否显示
二、导入数据,并查看
data = sns.load_dataset(‘tips‘) data.head()
#字段含义分别是:总消费,小费,性别,是否吸烟,周末,时间,几个人
三、作图(matplotlib)
1、折线图
fig,ax = plt.subplots() #fig主要设置一些全局的变量,而ax主要负责画图 ax.plot(data.index,data[‘total_bill‘]) fig.set_size_inches(12,6) #重新设置大小 plt.title(‘折线图标题‘,fontsize=22) #标题,更改字体大小 ax.set_xlabel(‘X轴‘,fontsize=18) #设置x轴,y轴的标题 ax.set_ylabel(‘Y轴‘,fontsize=18) plt.yticks(fontsize=14) #刻度字体大小 plt.xticks(fontsize=14) plt.legend([‘标签‘],fontsize=15) #标签内容字体大小 plt.savefig(‘折线图‘,dpi=100) #保存图片,可以设置dpi
2、柱形图
data_bar1 = data[‘tip‘].groupby(data[‘day‘]).mean() #统计数据,按照星期来分组 data_bar2 = data[‘tip‘].groupby(data[‘time‘]).mean() #统计数据,按照星期来分组 error = data[‘tip‘].groupby(data[‘time‘]).std() fig,ax = plt.subplots(1,2) #画出两个区域 ax[0].bar(data_bar1.index,data_bar1.values) #第一个区域怎么画 ax[0].set_xlabel(‘小费星期关系图‘,fontsize=16) ax[0].set_ylabel(‘小费‘,fontsize=16) ax[0].legend([‘星期‘]) ax[0].set_ylim(0,4) #设置Y轴最大最小值 ax[1].bar(data_bar2.index,data_bar2.values) #第二个区域怎么画 ax[1].errorbar(data_bar2.index,data_bar2.values,yerr = error,ls = ‘none‘,color=‘#96CDCD‘,lw=6) #加入方差图 ax[1].legend([‘午晚餐‘]) ax[1].set_xlabel(‘小费午晚餐关系图‘,fontsize=16) ax[1].set_xticklabels([‘晚餐‘,‘午餐‘]) #更改坐标轴的名称 ax[1].set_ylim(0,4) #设置Y轴最大最小值 fig.set_size_inches(12,6) #设置整个图的大小 plt.savefig(‘柱形图‘,dpi=100) #保存图片,可以设置dpi
3、横轴的柱形图
data_barh = data[‘tip‘].groupby(data[‘sex‘]).mean() #统计数据,男女小费 fig,ax = plt.subplots() ax.barh(data_barh.index,data_barh.values,0.4) #0.4是宽度 fig.set_size_inches(6,2) plt.title(‘横着的‘) ax.set_ylabel(‘性别‘,fontsize=20) ax.set_xlabel(‘小费‘,fontsize = 20) plt.xticks(fontsize=14) ax.set_yticklabels([‘男性‘,‘女性‘])
4、饼图
data_pie = data[‘size‘].groupby(data[‘size‘]).size() fig,ax = plt.subplots() ax.pie(data_pir,autopct=‘%1.1f%%‘,labels=data_pie.index,colors = [‘#B0E0E6‘,‘#B0C4DE‘,‘#A6A6A6‘,‘#FF3E96‘,‘#FFB5C5‘,‘#FFEBCD‘]) #数据源,显示的数值,显示标签,颜色 fig.set_size_inches(8,8) #如果两个数字不相等会变成椭圆
5、散点图
fig,ax = plt.subplots() ax.scatter(data[‘tip‘],data[‘total_bill‘]) fig.set_size_inches(8,6) ax.set_xlabel(‘小费‘,fontsize=18) ax.set_ylabel(‘总消费‘,fontsize=18) plt.xticks(fontsize=12) plt.yticks(fontsize=12)
6、几个区域的画图方法(一种是用上面的柱形图那种方法 fig,ax = plt.subplots(1,2),另一种是下面的这种,这种可以自定义占据的空格数)
fig = plt.figure() ax1 = plt.subplot2grid((2,3),(0,0)) ax1.bar(data_bar.index,data_bar.values) fig.set_size_inches(12,6) ax2 = plt.subplot2grid((2,3),(0,1),colspan=2)#占据几个空额,也可以是rowspan,一个是横的,一个是竖的 ax2.scatter(data[‘tip‘],data[‘total_bill‘]) ax3 = plt.subplot2grid((2,3),(1,0)) ax3.barh(data_barh.index,data_barh.values)
7、两根柱形图对比
fig,ax = plt.subplots() ax.bar(np.arange(4),data_bar.values,0.3) #横坐标先用数字代替 ax.bar(np.arange(4)+0.3,data_bar.values*2,0.3) #偏移一定量 ax.set_xticks(np.arange(4)+0.15) #重新设置x轴的位置 ax.set_xticklabels(data_bar.index) #重新设置名称
四、作图(seaborn)