Python可视化库-Matplotlib使用总结
在做完数据分析后,有时候需要将分析结果一目了然地展示出来,此时便离不开Python可视化工具,Matplotlib是Python中的一个2D绘图工具,是另外一个绘图工具seaborn的基础包
先总结下绘制子图的步骤:
1.确定绘制的图形形状(如折线图/条状图/柱状图/饼图/散点图等)
2.填充x/y轴的数据
3.图形细节调整(这里可以做很多调整,如x/y轴文字参数说明,颜色/线粗/柱状粗度,x/y轴文字角度等)
4.显示图像(调用show())
总结下一个区域同时绘制多个子图的步骤
1.确定绘图区域大小
2.确定每个子图在绘图区域的位置
3.绘制每个子图(步骤如上)
4.显示图像(调用show())
绘制折线图
需求:根据一张美国的失业率数据,绘制出1948年12个月中的失业率折线图,其中x轴表示月份,y轴表示失业率
数据如下
DATE VALUE
0 1948-01-01 3.4
1 1948-02-01 3.8
2 1948-03-01 4.0
3 1948-04-01 3.9
4 1948-05-01 3.5
5 1948-06-01 3.6
6 1948-07-01 3.6
7 1948-08-01 3.9
8 1948-09-01 3.8
9 1948-10-01 3.7
10 1948-11-01 3.8
11 1948-12-01 4.0
unrate.csv
上代码
#导包
import pandas as pd
import matplotlib.pyplot as plt
# 读取本地的csv文件,里面的数据为美国失业率数据,得到的数据为DataFrame类型
unrate = pd.read_csv('unrate.csv')
# 使用pd中的pd.to_datetime函数将DATE列的字符串类型数据转换成pd中的标准时间格式
unrate['DATE'] = pd.to_datetime(unrate['DATE'])
# 取出前面12条样本数据
first_twelve = unrate[0:12]
# 填充数据并绘制折线图,第一个参数为x轴数据,第二个参数为y轴数据
plt.plot(first_twelve['DATE'], first_twelve['VALUE'])
# 将x轴下面文字旋转90度
plt.xticks(rotation=90)
# 设置x轴的标签
plt.xlabel('Month')
# 设置y轴的标签
plt.ylabel('Unemployment Rate')
# 设置图标名称
plt.title('Monthly Unemployment Trends, 1948')
# 显示图像
plt.show()
图像如图所示:

在一个区域绘制多个子图
有时候,需要将多张图像在一块区域显示,方便对比
上代码:
# 导包
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
# 确定总绘图区宽和高分别为都是3x6
fig = plt.figure(figsize=(3, 6))
# 添加第一个子图,并且确定在总绘图区域的位置,add_subpolt(2,1,1),前两个参数参数2,1表示将总绘图区域划分为两行1列(跟矩阵表示很像)
# 第3个参数表示该子图占总区域的第一个位置.注(将总区域分成2行1列后,位置顺序从上到下,从左到右,从1开始递增)
ax1 = fig.add_subplot(2,1,1)
ax2 = fig.add_subplot(2,1,2)
# 绘制子图
ax1.plot(np.random.randint(1,5,5), np.arange(5))
ax2.plot(np.arange(10)*3, np.arange(10))
# 显示图像
plt.show()
图像如图所示

在一张图中同时绘制多条曲线
需求:在同一张图,同时将1948到1952年的失业率展示出来,曲线使用不同的颜色区分
上代码
#绘制5年内的失业率曲线图
#导包
import numpy as np
import pandas as pd
import matplotlib.pypolt as plt
# 读取本地数据
unrate=pd.read_csv('UNRATE.csv')
# 将字符串时间转换成pd中的时间格式
unrate['DATE']=pd.to_datetime(unrate['DATE'])
# 将年份时间转换成月份时间,并新建列存起来,因为需求需要将5年内的当年12个月内的失业率展示出来,此时再用年时间作为x轴下标就不合适了
unrate['MONTH']=unrate['DATE'].dt.month
# 设置5个颜色数组,分别表示5条曲线颜色
colors=['red','green','blue','black','yellow']
# 设置总绘图区域大小,需要在调用绘图函数之前调用才有效果
plt.figure(figsize=(10,6))
#遍历5次
for i in range(5):
#取出当年12个月的数据
data12=unrate[12*i:12*(i+1)]
#x轴数据
data_x=data12['MONTH']
#y轴数据
data_y=data12['VALUE']
#当年的曲线的标签
label=str(1948+i)
# 绘制当年的曲线图
plt.plot(data_x,data_y,c=colors[i],label=label)
# 设置x轴标签
plt.xlabel('MONTH')
# 设置y轴标签
plt.ylabel('unrate')
# 设置图标名称
plt.title('US-unrae,1948-1952')
# 设置曲线标签说明,loc='best'表示自己选择合适的位置来摆放
plt.legend(loc='best')
# 显示图像
plt.show()
得到的图像:

绘制条形图
直方图和条形图的区别:由于分组数据具有连续性，直方图中的各矩形通常是连续排列，而条形图则是分开排列。此外直方图的高度表示各小组内数据个数，而条形图高度表示某项目内的数据个数。
fandango_scores.csv
上代码:
# 导包
import pandas as pd
import matplotlib.pyplot as plt
from numpy import arange
# 读取电影评分数据
reviews = pd.read_csv('fandango_scores.csv')
# 取出需要展示的列名
cols = ['FILM', 'RT_user_norm', 'Metacritic_user_nom', 'IMDB_norm', 'Fandango_Ratingvalue', 'Fandango_Stars']
# 取出需要展示的样本数据
norm_reviews = reviews[cols]
# 评分数据的列名
num_cols = ['RT_user_norm', 'Metacritic_user_nom', 'IMDB_norm', 'Fandango_Ratingvalue', 'Fandango_Stars']
# 取出第一部电影的评分样本数据
bar_heights = norm_reviews.loc[0, num_cols].values
# 条形图的位置
bar_positions = arange(5) + 0.75
tick_positions = range(1,6)
fig, ax = plt.subplots()
# 绘制条形图,第一个参数为x轴数据,第二个参数为y轴数据,第三个参数为每个条形的宽度
ax.bar(bar_positions, bar_heights,0.5)
# 设置x轴标签的位置
ax.set_xticks(tick_positions)
# 设置x轴标签的名字和角度
ax.set_xticklabels(num_cols, rotation=45)
# 设置x轴标签
ax.set_xlabel('Rating Source')
# y轴标签
ax.set_ylabel('Average Rating')
# 标题
ax.set_title('Average User Rating For Avengers: Age of Ultron (2015)')
plt.show()
绘制的图片:

绘制直方图
直方图一般用来统计一定范围内的数据
上代码
import pandas as pd
import matplotlib.pyplot as plt
reviews = pd.read_csv('fandango_scores.csv')
cols = ['FILM', 'RT_user_norm', 'Metacritic_user_nom', 'IMDB_norm', 'Fandango_Ratingvalue']
norm_reviews = reviews[cols]
# 对评分进行数量统计
fandango_distribution = norm_reviews['Fandango_Ratingvalue'].value_counts()
# 排序索引
fandango_distribution = fandango_distribution.sort_index()
imdb_distribution = norm_reviews['IMDB_norm'].value_counts()
imdb_distribution = imdb_distribution.sort_index()
fig, ax = plt.subplots()
# 绘制直方图
ax.hist(norm_reviews['Fandango_Ratingvalue'],bins=20)
plt.show()
绘制所得图像:

绘制散点图
上代码:
# 导包
import pandas as pd
import matplotlib.pyplot as plt
from numpy import arange
# 读取电影评分数据
norm_reviews = pd.read_csv('fandango_scores.csv')
# 取出需要展示的列名
cols = ['FILM', 'RT_user_norm', 'Metacritic_user_nom', 'IMDB_norm', 'Fandango_Ratingvalue', 'Fandango_Stars']
# 取出需要展示的样本数据
norm_reviews = reviews[cols]
fig, ax = plt.subplots()
#绘制散点图
ax.scatter(norm_reviews['Fandango_Ratingvalue'], norm_reviews['RT_user_norm'])
ax.set_xlabel('Fandango')
ax.set_ylabel('Rotten Tomatoes')
plt.show()
绘制所得图片:

绘制盒图
箱形图（英文：Box-plot），又称为盒须图、盒式图、盒状图或箱线图，是一种用作显示一组数据分散情况资料的统计图。因型状如箱子而得名。在各种领域也经常被使用，常见于品质管理。不过作法相对较繁琐。它能显示出一组数据的最大值、最小值、中位数、下四分位数及上四分位数。

上代码:
import pandas as pd
import matplotlib.pyplot as plt
reviews = pd.read_csv('fandango_scores.csv')
cols = ['FILM', 'RT_user_norm', 'Metacritic_user_nom', 'IMDB_norm', 'Fandango_Ratingvalue']
norm_reviews = reviews[cols]
num_cols = ['RT_user_norm', 'Metacritic_user_nom', 'IMDB_norm', 'Fandango_Ratingvalue']
fig, ax = plt.subplots()
#绘制盒图
ax.boxplot(norm_reviews[num_cols].values)
ax.set_xticklabels(num_cols, rotation=90)
ax.set_ylim(0,5)
plt.show()
绘制所得图片:

