Python数据分析之Matplotlib的常用操作总结

顺势而为47

目录

• 使用准备
  
• 1、简单的绘制图像
  
• 2、视图面板的常用操作
  
• 3、样式及各类常用修饰属性
  
• 4、legend图例的使用
  
• 5、添加文字等描述
  
• 6、不同类型图像的绘制
  
• 总结
  

使用准备

使用matplotlib需引入：

1. import matplotlib.pyplot as plt

复制代码

通常2会配合着numpy使用，numpy引入：

1. import numpy as np

复制代码

1、简单的绘制图像

1. def matplotlib_draw():
   
2.     # 从-1到1生成100个点，包括最后一个点，默认为不包括最后一个点
   
3.     x = np.linspace(-1, 1, 100, endpoint=True)
   
4. 
   
5.     y = 2 * x + 1
   
6.     plt.plot(x, y)  # plot将信息传入图中
   
7.     plt.show()  # 展示图片

复制代码

2、视图面板的常用操作

1. def matplotlib_figure():
   
2.     x = np.linspace(-1, 1, 100)
   
3.     y1 = 2 * x + 1
   
4.     y2 = x ** 2  # 平方
   
5. 
   
6.     plt.figure()  # figure是视图面板
   
7.     plt.plot(x, y1)
   
8. 
   
9.     # 这里再创建一个视图面板，最后会生成两张图，figure只绘制范围以下的部分
   
10.     plt.figure(figsize=(4, 4))  # 设置视图长宽
    
11.     plt.plot(x, y2)
    
12. 
    
13.     plt.show()

复制代码

3、样式及各类常用修饰属性

1. def matplotlib_style():
   
2.     x = np.linspace(-3, 3, 100)
   
3.     y1 = 2 * x + 1
   
4.     y2 = x ** 2  # 平方
   
5. 
   
6.     # 限制xy输出图像的范围
   
7.     plt.xlim((-1, 2))  # 限制x的范围
   
8.     plt.ylim((-2, 3))  # 限制y的范围
   
9. 
   
10.     # xy描述
    
11.     plt.xlabel('I am X')
    
12.     plt.ylabel('I am Y')
    
13. 
    
14.     # 设置xy刻度值
    
15.     # 从-2到2上取11个点,最后生成一个一维数组
    
16.     new_sticks = np.linspace(-2, 2, 11)
    
17.     plt.xticks(new_sticks)
    
18.     # 使用文字代替数字刻度
    
19.     plt.yticks([-1, 0, 1, 2, 3], ['level1', 'level2', 'level3', 'level4', 'level5'])
    
20. 
    
21.     # 获取坐标轴 gca get current axis
    
22.     ax = plt.gca()
    
23.     ax.spines['right'].set_color('red')  # 设置右边框为红色
    
24.     ax.spines['top'].set_color('none')  # 设置顶部边框为没有颜色，即无边框
    
25. 
    
26.     # 把x轴的刻度设置为'bottom'
    
27.     ax.xaxis.set_ticks_position('bottom')
    
28.     # 把y轴的刻度设置为'left'
    
29.     ax.yaxis.set_ticks_position('left')
    
30.     # 设置xy轴的位置,以下测试xy轴相交于(1,0)
    
31.     # bottom对应到0点
    
32.     ax.spines['bottom'].set_position(('data', 0))
    
33.     # left对应到1点
    
34.     ax.spines['left'].set_position(('data', 1))  # y轴会与1刻度对齐
    
35. 
    
36.     # 颜色、线宽、实线：'-',虚线：'--',alpha表示透明度
    
37.     plt.plot(x, y1, color="red", linewidth=1.0, linestyle='--', alpha=0.5)
    
38.     plt.plot(x, y2, color="blue", linewidth=5.0, linestyle='-')
    
39.     plt.show()  # 这里没有设置figure那么两个线图就会放到一个视图里

复制代码

4、legend图例的使用

1. def matplotlib_legend():
   
2.     x = np.linspace(-3, 3, 100)
   
3.     y1 = 2 * x + 1
   
4.     y2 = x ** 2  # 平方
   
5. 
   
6.     l1, = plt.plot(x, y1, color="red", linewidth=1.0, linestyle='--', alpha=0.5)
   
7.     l2, = plt.plot(x, y2, color="blue", linewidth=5.0, linestyle='-')
   
8.     # handles里面传入要产生图例的关系线，labels中传入对应的名称，
   
9.     # loc='best'表示自动选择最好的位置放置图例
   
10.     plt.legend(handles=[l1, l2], labels=['test1', 'test2'], loc='best')
    
11.     plt.show()

复制代码

5、添加文字等描述

1. def matplotlib_describe():
   
2.     x = np.linspace(-3, 3, 100)
   
3.     y = 2 * x + 1
   
4.     plt.plot(x, y, color="red", linewidth=1.0, linestyle='-')
   
5. 
   
6.     # 画点,s表示点的大小
   
7.     x0 = 0.5
   
8.     y0 = 2 * x0 + 1
   
9.     plt.scatter(x0, y0, s=50, color='b')
   
10. 
    
11.     # 画虚线,
    
12.     # k代表黑色，--代表虚线,lw线宽
    
13.     # 表示重(x0,y0)到(x0,-4)画线
    
14.     plt.plot([x0, x0], [y0, -4], 'k--', lw=2)
    
15.     # 标注,xytext:位置，textcoords设置起始位置，arrowprops设置箭头，connectionstyle设置弧度
    
16.     plt.annotate(r'$2x+1=%s$' % y0, xy=(x0, y0), xytext=(+30, -30),
    
17.                  textcoords="offset points", fontsize=16,
    
18.                  arrowprops=dict(arrowstyle='->', connectionstyle='arc3,rad=.2'))
    
19. 
    
20.     # 文字描述
    
21.     plt.text(-3, 3, r'$this\ is\ the\ text$', fontdict={'size': '16', 'color': 'r'})
    
22. 
    
23.     plt.show()

复制代码

6、不同类型图像的绘制

（1）scatter绘制散点图：

1. def matplotlib_scatter():
   
2.     plt.figure()
   
3.     plt.scatter(np.arange(5), np.arange(5))  # 安排两个0到4的数组绘制
   
4. 
   
5.     x = np.random.normal(0, 1, 500)  # 正态分布的500个数
   
6.     y = np.random.normal(0, 1, 500)
   
7. 
   
8.     plt.figure()
   
9.     plt.scatter(x, y, s=50, c='b', alpha=0.5)
   
10. 
    
11.     plt.show()

复制代码

（2）bar绘制直方图：

1. def matplotlib_bar():
   
2.     x = np.arange(10)
   
3.     y = 2 ** x + 10
   
4.     # facecolor块的颜色，edgecolor块边框的颜色
   
5.     plt.bar(x, y, facecolor='#9999ff', edgecolor='white')
   
6.     # 设置数值位置
   
7.     for x, y in zip(x, y):  # zip将x和y结合在一起
   
8.         plt.text(x + 0.4, y, "%.2f" % y, ha='center', va='bottom')
   
9. 
   
10.     plt.show()

复制代码

（3）contour轮廓图:

1. def matplotlib_contours():
   
2.     def f(a, b):
   
3.         return (1 - a / 2 + a ** 5 + b ** 3) * np.exp(-a ** 2 - b ** 2)
   
4. 
   
5.     x = np.linspace(-3, 3, 100)
   
6.     y = np.linspace(-3, 3, 100)
   
7. 
   
8.     X, Y = np.meshgrid(x, y)  # 将x和y传入一个网格中
   
9.     # 8表示条形线的数量，数量越多越密集
   
10.     plt.contourf(X, Y, f(X, Y), 8, alpha=0.75, cmap=plt.cm.hot)  # cmap代表图的颜色
    
11. 
    
12.     C = plt.contour(X, Y, f(X, Y), 8, color='black', linewidth=.5)
    
13.     plt.clabel(C, inline=True, fontsize=10)
    
14. 
    
15.     plt.xticks(())
    
16.     plt.yticks(())
    
17.     plt.show()

复制代码

（4）3D图：
3D图绘制需额外再引入依赖：

1. from mpl_toolkits.mplot3d import Axes3D

复制代码

1. def matplotlib_Axes3D():
   
2.     fig = plt.figure()  # 创建绘图面版环境
   
3.     ax = Axes3D(fig)  # 将环境配置进去
   
4. 
   
5.     x = np.arange(-4, 4, 0.25)
   
6.     y = np.arange(-4, 4, 0.25)
   
7.     X, Y = np.meshgrid(x, y)  
   
8.     R = np.sqrt(X ** 2 + Y ** 2)
   
9.     Z = np.sin(R)
   
10. 
    
11.     # stride控制色块大小
    
12.     ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap=plt.get_cmap('rainbow'))
    
13.     ax.contourf(X, Y, Z, zdir='z', offset=-2, cmap='rainbow')
    
14.     ax.set_zlim(-2, 2)
    
15. 
    
16.     plt.show()

复制代码

（5）subplot子图绘制：

1. def matplotlib_subplot():
   
2.     plt.figure()  # 生成绘图面板
   
3.     plt.subplot(2, 1, 1)  # 两行1列绘图位置的第1个位置
   
4.     plt.plot([0, 1], [0, 1])  # 绘制从(0,0)绘制到(1,1)的图像
   
5.     plt.subplot(2, 3, 4)  # 两行3列绘图位置的第4个位置
   
6.     plt.plot([0, 1], [0, 1])  # 绘制从(0,0)绘制到(1,1)的图像
   
7.     plt.subplot(2, 3, 5)  # 两行3列绘图位置的第5个位置
   
8.     plt.plot([0, 1], [0, 1])  # 绘制从(0,0)绘制到(1,1)的图像
   
9.     plt.subplot(2, 3, 6)  # 两行3列绘图位置的第6个位置
   
10.     plt.plot([0, 1], [0, 1])  # 绘制从(0,0)绘制到(1,1)的图像
    
11. 
    
12.     plt.show()

复制代码

（6）animation动图绘制
需额外导入依赖：

1. from matplotlib import animation

复制代码

1. # ipython里运行可以看到动态效果
   
2. def matplotlib_animation():
   
3.     fig, ax = plt.subplots()
   
4. 
   
5.     x = np.arange(0, 2 * np.pi, 0.01)
   
6.     line, = ax.plot(x, np.sin(x))
   
7. 
   
8.     def animate(i):
   
9.         line.set_ydata(np.sin(x + i / 10))
   
10.         return line,
    
11. 
    
12.     def init():
    
13.         line.set_ydata(np.sin(x))
    
14.         return line,
    
15. 
    
16.     ani = animation.FuncAnimation(fig=fig, func=animate, init_func=init, interval=20)
    
17. 
    
18.     plt.show()

复制代码

附：直方图代码实现

1. import numpy as np
   
2. import matplotlib.pyplot as plt
   
3. 
   
4. np.random.seed(1)
   
5. # 产生30个学生身高数据
   
6. hight = np.random.randint(low=140, high=190, size=30)
   
7. print("身高数据", hight)
   
8. 
   
9. # 绘制直方图 plt.hist
   
10. 
    
11. # 参数1：要统计的数据； 参数2：区间信息
    
12. 
    
13. # 区间信息有默认值 bins =10  分10组
    
14. # bins = [140, 145, 160, 170, 190]
    
15. # 除了最后一个 都是前闭后开；最后一组是前闭后闭
    
16. # [140,145) [145,160) [160,170) [170,190]
    
17. 
    
18. bins = [140, 180, 190]
    
19. 
    
20. cnt, bins_info, _ = plt.hist(hight,
    
21.                              bins=10,
    
22.                              # bins=bins,
    
23.                              edgecolor='w'  # 柱子的边缘颜色 白色
    
24.                              )
    
25. # 直方图的返回值有3部分内容
    
26. # 1. 每个区间的数据量
    
27. # 2. 区间信息
    
28. # 3. 区间内数据数据信息 是个对象 不能直接查看
    
29. # print("直方图的返回值", out)
    
30. 
    
31. # cnt, bins_info, _ = out
    
32. 
    
33. 
    
34. # 修改x轴刻度
    
35. plt.xticks(bins_info)
    
36. 
    
37. # 增加网格线
    
38. # 参数1：b bool类型 是否增加网格线
    
39. # 参数 axis 网格线 垂直于 哪个轴
    
40. plt.grid(b=True,
    
41.          axis='y',
    
42.          # axis='both'
    
43.          alpha=0.3
    
44.          )
    
45. 
    
46. # 增加标注信息 plt.text
    
47. print("区间信息", bins_info)
    
48. print("区间数据量", cnt)
    
49. 
    
50. bins_info_v2 = (bins_info[:-1] + bins_info[1:]) / 2
    
51. for i, j in zip(bins_info_v2, cnt):
    
52.     # print(i, j)
    
53.     plt.text(i, j + 0.4, j,
    
54.              horizontalalignment='center',  # 水平居中
    
55.              verticalalignment='center',  # 垂直居中
    
56.              )
    
57. 
    
58. # 调整y轴刻度
    
59. plt.yticks(np.arange(0, 20, 2))
    
60. 
    
61. plt.show()

复制代码

更多见官方文档：教程 | Matplotlib 中文

总结

到此这篇关于Python数据分析之Matplotlib常用操作的文章就介绍到这了,更多相关Python Matplotlib常用操作内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
                                                        
免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！