2022年 11月 13日

Python直接使用plot()函数画图

目录

一、plot()函数的认识

二、plot()函数基本运用

 三、plot()函数数据可视化画图以及图元基本参数设置


一、plot()函数的认识

在使用Python进行数据可视化编程中matplotlib库是我们用来对数据进行画图常用的第三方库。其中含有各类函数也就是不同类型的图形,要使用matplotlib库中的函数就需要了解函数所需要数据的格式,这也是我们学习matplotlib库的重点。

直接使用plot()函数画图,是对于一般的简单数据。我们可以采用直接调用plot()函数对列表数据进行直接画图。初期学习直接使用plot()函数能便于我们对后面图形学习奠定函数的参数及基础。

matplotlib图的组成:

  • Figure (画布)
  • Axes (坐标系)
  • Axis (坐标轴)
  • 图形(plot(),scatter(),bar(),…)
  • Title, Labels, ……

直接是用plot()函数画图如法如下:

plt.plot(x, y, fmt='xxx', linestyle=, marker=, color=, linewidth=, markersize=, label=, )

其中x,y代表横纵坐标,fmt = ‘#color#linestyle#marker’即代表各类参数。

(1)linestyle:此字段是线的样式,参数形式:字符串

linestyle(线的样式)
linestyle参数 线形
‘-‘ 实线
‘–‘ 虚线
‘-.’ 点划线
‘:’ 点虚线
‘ ‘ 无线

(2)linewidth:此参数是线的粗细,粗细程度和所定数值大小有关,参数形式:数值

(3)marker:点的样式,字符串

marker(点的样式)
marker 标记点
‘.’
‘,’ 像素
‘^’ ‘v’ ‘>’ ‘<‘ 上下左右三角形
‘1’ ‘2’ ‘3’ ‘4’ 上下左右三叉线
‘o’ 圆形
‘s’ ‘D’ 方形
‘p’ 五边形
‘h’ ‘H’ 六边形
‘*’ 五角星
‘+’ ‘x’ 十字交叉
‘_’ 横线

(4)markersize:点的大小,参数形式:数值

(5)color:调节线条还有点的颜色 ,字符串,参数形式字符串

color(点、线颜色)
字符串 color
‘r’
‘g’ 绿
‘b’
‘y’
‘c’
‘m’
‘k’
‘w’

此处颜色参数还可以有二进制,十进制等表示方法,同时对于颜色,RGB是三原色

(6)label:图例,legend文字

二、plot()函数基本运用

使用plot()函数时需要导入对应的库,导入库后我们在未有数据的情况下直接画图,直接画图会隐式创建Figure,Axes对象。

  1. import matplotlib.pyplot as plt
  2. plt.plot()

 下面通过构造数据绘制简单图形

首先数据构造,设置参数,参数也可以在将数据填入plot()函数的时候设置。

  1. # 导入包
  2. import matplotlib.pyplot as plt
  3. import numpy as np
  4. # 构造数据
  5. # 位置 (2维:x,y一一对应)
  6. x = np.linspace(0, 2 * np.pi, 200) # 从0到2pi的200个值
  7. y = np.sin(x) # 从sin(0)到sin(2pi)的200个值
  8. # 颜色(0维)
  9. c = 'red'
  10. c = 'r'
  11. c = '#FF0000'
  12. # 大小(0维): 线宽
  13. lw = 1

画出图形

  1. # 生成一个Figure画布和一个Axes坐标系
  2. fig, ax = plt.subplots()
  3. # 在生成的坐标系下画折线图
  4. ax.plot(x, y, c, linewidth=lw)
  5. # 显示图形
  6. plt.show()

图形展示:

给定两组数据,建立y与x的关系试,使用plot函数进行画图,本次画图线条选用点虚线形式,粗细选用1,点选用方形,点大小选用值为10,图例为‘1234’

  1. import matplotlib.pyplot as plt
  2. x = [1,2,3]
  3. y = [1,2,3]
  4. y = x
  5. plt.plot(x,y,linestyle=':', linewidth=1, marker='d', markersize=10, label='1234')
  6. plt.legend()

作出图片如下;

 下面我们引用numpy的linspace函数生创建均匀分布序列,然后对x,y建立数值关系,由此来创建图画。

  1. import matplotlib.pyplot as plt
  2. import numpy as np
  3. x = np.linspace(-100,100,10)
  4. y = x**2 + 2*x +1
  5. plt.plot(x,y,'g-.o')

作出如下图案,由此可见,我们对于图形的设置方面,在我们熟练以后如果没有粗细的设置可以直接缩减再一个字符串里面

以上都是简单图形的讲解,我们现在通过一个简单的对数据DataFrame进行作图,在往后的数据可视化中我们需要对数据进行处理后再进行可视化。下面我们通过正、余弦函数进行作图。

  1. #导入包
  2. import matplotlib.pyplot as plt
  3. import numpy as np
  4. import pandas as pd
  5. #使用linspace()方法构成数据
  6. x = np.linspace(0, 2 * np.pi, 50) #
  7. y1 = np.sin(x)
  8. y2 = np.cos(x)
  9. #转化数据形式
  10. df = pd.DataFrame([x,y1,y2]).T
  11. #对列重新命名
  12. df.columns = ['x','sin(x)','cos(x)']
  13. #数据写入图像,命名图例
  14. plt.plot(df['x'],df['sin(x)'],label='sin(x)')
  15. plt.plot(df['x'],df['cos(x)'],label='cos(x)')
  16. plt.legend()

我们通过numpy的linspace方法生成数据再通过pandas对数据进行DataFrame化再带入plot()函数,此处需要讲的就是图例的命名方法,通过在函数中写入label参数,确定图例的标签,再通过legend()函数生成图例,在后续的学习中也会讲到图例的位置、形式等的运用。

 

 三、plot()函数数据可视化画图以及图元基本参数设置

通过绘制世界人口变化曲线图了解基本图元参数设置,本次绘图过程主要是先通过对人口数据导入,了解数据构造,再进配置画图参数最后完成图形的制作,其中基本的图元参数用于别的图形也适用,在这儿学会了我们只需要了解数据结构,构造成图形所要的数据结构就可以就行画出自己想要的图形。

首先进行数据导入,了解数据结构形式。为了学习方便,选用jupyter notebook进行可视化图形讲解。

  1. import pandas as pd
  2. datafile = r'world_population.txt' # 打开文件
  3. df = pd.read_csv(datafile) #读取数据
  4. df.head()#展示前面部分数据

以下就是基本的数据样式,由年份和人口数量组成

 这里做了基本的图元设计,也就是对于画布的设置,前面我们所学函数参数都是对于图形中间的设置,我们构成一个可视化界面是通过画布+画中图形样式组成一个完整的可视化界面。

画布界面有画布大小,画布像素,画布界面,画布边框等设置。

  1. import matplotlib.pyplot as plt
  2. # 画布
  3. fig = plt.figure(figsize=(6,4), # inches
  4. dpi=120, # dot-per-inch
  5. facecolor='#BBBBBB',
  6. frameon=True, # 画布边框
  7. )
  8. plt.plot(df['year'],df['population'])
  9. # 标题
  10. plt.title("1960-2009 World Population")

构成一个完整的可视化图像除了图例还有图像的标题,我们可以通过title()方法设置英文标题,中文标题要通过以下代码才能实现,因此我们如果是做中文项目在导入包以后就可以写上设置中文代码的代码串。

  1. # 设置中文字体
  2. plt.rcParams['font.sans-serif'] = 'SimHei' # 设置字体为简黑(SimHei)
  3. plt.rcParams['font.sans-serif'] = 'FangSong' # 设置字体为仿宋(FangSong)

 当然,除了这种比较简单的图形之外我们还能对图形进行优化设置,将数据显示的更加的精美和美观,对图形优化便于实际报告中的演示也是我们现在必不可少的的一环。

  1. import matplotlib.pyplot as plt
  2. # 设置中文字体
  3. plt.rcParams['axes.unicode_minus'] = False # 不使用中文减号
  4. plt.rcParams['font.sans-serif'] = 'FangSong' # 设置字体为仿宋(FangSong)
  5. # 画布
  6. fig = plt.figure(figsize=(6,4), # inches
  7. dpi=120, # dot-per-inch
  8. facecolor='#BBBBBB',
  9. frameon=True, # 画布边框
  10. )
  11. plt.plot(df['year'],df['population'],'b:o',label='人口数')
  12. # 中文标题
  13. plt.title("1960-2009 世界人口")
  14. # 字体字典
  15. font_dict=dict(fontsize=8,
  16. color='k',
  17. family='SimHei',
  18. weight='light',
  19. style='italic',
  20. )
  21. # X轴标签
  22. plt.xlabel("年份", loc='center', fontdict=font_dict) # loc: 左中右 left center right
  23. # Y轴标签
  24. plt.ylabel("人口数",loc='top', fontdict=font_dict) # loc: 上中下 top center bottom
  25. # X轴范围
  26. plt.xlim((2000,2010)) # X轴的起点和终点
  27. # Y轴范围
  28. plt.ylim(6e9,7e9) # Y轴的起点和终点
  29. # X轴刻度
  30. plt.xticks(np.arange(2000,2011))
  31. # X轴刻度
  32. plt.yticks(np.arange(6e9,7e9+1e8,1e8))
  33. # 图例
  34. plt.legend()
  35. # plt.legend(labels=['人口'])
  36. # 网格线
  37. plt.grid(axis='y') # axis: 'both','x','y'

 上述代码,对x轴,y轴的刻度、标签、字体进行定义,对图例、网格线等也做出了参数的设置,最后做出的图形如下图: