Python3快速入门（十六）——Mat
Python3快速入门（十六）——Matplotlib绘图
一、Matplotlib简介
1、Matplotlib简介
Matplotlib是 Python 2D-绘图领域使用最广泛的套件，可以简易地将数据图形化，并且提供多样化的输出格式。 matplotlib有两个接口，一个是状态机层的接口，通过pyplot模块来进行管理；一个是面向对象的接口，通过pylab模块将所有的功能函数全部导入其单独的命名空间内。
2、Matplotlib安装
使用conda安装如下：
conda install matplotlib
二、Matplotlib图表结构
1、Matplotlib图表结构简介
Matplotlib基本图表结构包括坐标轴（X轴、Y轴）、坐标轴标签（axisLabel）、 坐标轴刻度（tick）、坐标轴刻度标签（tick label）、绘图区（axes）、画布（figure）。
Python3快速入门（十六）——Matplotlib绘图
2、Figure
Figure代表一个绘制面板，其中可以包涵多个Axes（即多个图表）。 Axes表示一个图表 ，一个Axes包涵：titlek、xaxis、yaxis。 为了支持pylab中的gca()等函数，Figure对象内部保存有当前轴的信息，因此不建议直接对Figure.axes属性进行列表操作，而应该使用add_subplot, add_axes, delaxes等方法进行添加和删除操作。
# -*- coding=utf-8 -*-
import matplotlib.pyplot as plt
import numpy as np
if __name__ == "__main__":
fig = plt.figure()
ax1 = fig.add_axes([0.1, 0.45, 0.8, 0.5])
ax2 = fig.add_axes([0.1, 0.1, 0.8, 0.2])
x1 = np.linspace(0.0, 5.0)
x2 = np.linspace(0.0, 3.0)
y1 = np.cos(2 * np.pi * x1) * np.exp(-x1)
y2 = np.cos(2 * np.pi * x2)
ax1.patch.set_facecolor("green")
ax1.grid(True)
line1 = ax1.plot(x1, y1, 'yo-')
line2 = ax2.plot(x2, y2, 'r.-')
plt.show()
Python3快速入门（十六）——Matplotlib绘图
3、网格线设置
网格线设置
plt.grid(color='r',linestyle='-.')
axis：坐标轴，可选值为x，y
color:支持十六进制颜色
linestyle: –，-.，:
alpha：透明度，0——1
4、坐标轴设置
坐标轴范围设置
plt.axis([xmin,xmax,ymin,ymax])
也可以通过xlim(xmin,xmax)，ylim(xmin,xmax)方法设置坐标轴范围
# -*- coding=utf-8 -*-
import matplotlib.pyplot as plt
import numpy as np
if __name__ == "__main__":
x = np.arange(-10, 10, 0.1)
y = x ** 2
plt.plot(x, y,)
plt.axis([-10, 10, 0, 100])
plt.show()
Python3快速入门（十六）——Matplotlib绘图
关闭坐标轴
plt.axis('off')
5、画布设置
设置画布比例
plt.figure(figsize=(a,b))
a是x轴刻度比例，b是y轴刻度比例。
6、图例设置
图例设置有两种方法，一种是分别在plot函数中使用label参数指定，再调用plt.legend()方法显示图例；一种是直接在legend方法中传入字符串列表设置图例。
# -*- coding=utf-8 -*-
import matplotlib.pyplot as plt
import numpy as np
if __name__ == "__main__":
x = np.arange(-10, 10, 0.1)
y = x ** 2
plt.plot(x, y, label='y = x ** 2')
plt.legend()
plt.show()
Python3快速入门（十六）——Matplotlib绘图
使用legend函数设置图例时，参数如下： 图例名称列表：传递的图例名称列表必须与曲线绘制顺序一致。 loc：用于设置图例标签的位置，matplotlib预定义了多种数字表示的位置。 best：0，upper right：1，upper left：2，lower left：3，lower right：4，right：5，center left：6，center right：7，lower center：8，upper center：9，center：10，loc参数可以是2个元素的元组，表示图例左下角的坐标，[0,0] 左下，[0,1] 左上，[1,0] 右下，[1,1] 右上。 ncol：图例的列数
# -*- coding=utf-8 -*-
import matplotlib.pyplot as plt
import numpy as np
if __name__ == "__main__":
x1 = np.linspace(0, 2 * np.pi, 100)
y1 = np.sin(x1)
plt.plot(x1, y1)
x2 = x1 = np.linspace(0, 2 * np.pi, 100)
y2 = np.cos(x1)
plt.plot(x2, y2)
plt.legend(['sin(x)', 'cos(x)'], loc=0, ncol=1)
plt.show()
Python3快速入门（十六）——Matplotlib绘图
7、标题设置
标题设置可以使用plt.title()方法或ax.set_title()方法。
三、Matplotlib常见图表绘制
1、曲线图
抛物线绘制：
# -*- coding=utf-8 -*-
import matplotlib.pyplot as plt
import numpy as np
if __name__ == "__main__":
x = np.arange(-10, 10, 0.1)
y = x ** 2
plt.plot(x, y)
plt.show()
Python3快速入门（十六）——Matplotlib绘图
正弦曲线绘制：
# -*- coding=utf-8 -*-
import matplotlib.pyplot as plt
import numpy as np
if __name__ == "__main__":
x = np.linspace(0, 2 * np.pi, 100)
y = np.sin(x)
plt.plot(x, y)
plt.show()
Python3快速入门（十六）——Matplotlib绘图
多条曲线绘制： 多次调用plot函数可以在图上绘制多条曲线。
# -*- coding=utf-8 -*-
import matplotlib.pyplot as plt
import numpy as np
if __name__ == "__main__":
x1 = np.linspace(0, 2 * np.pi, 100)
y1 = np.sin(x1)
plt.plot(x1, y1)
x2 = x1 = np.linspace(0, 2 * np.pi, 100)
y2 = np.cos(x1)
plt.plot(x2, y2)
plt.show()
Python3快速入门（十六）——Matplotlib绘图
可以在一个plot函数中传入多对X,Y值，在一个图中绘制多个曲线。
# -*- coding=utf-8 -*-
import matplotlib.pyplot as plt
import numpy as np
if __name__ == "__main__":
x1 = np.linspace(0, 2 * np.pi, 100)
y1 = np.sin(x1)
x2 = x1 = np.linspace(0, 2 * np.pi, 100)
y2 = np.cos(x1)
plt.plot(x1, y1, x2, y2)
plt.show()2、直方图
# -*- coding=utf-8 -*-
import matplotlib.pyplot as plt
import numpy as np
if __name__ == "__main__":
x = np.random.randint(0, 100, 100)
bins = np.arange(0, 101, 10)
fig = plt.figure(figsize=(12, 6))
plt.subplot(1, 1, 1)
plt.hist(x, bins, color='b', alpha=0.6)
plt.show()
Python3快速入门（十六）——Matplotlib绘图
3、折线图
# -*- coding=utf-8 -*-
import matplotlib.pyplot as plt
import numpy as np
if __name__ == "__main__":
x = [1, 2, 3, 4, 5]
y = [2.3, 3.4, 1.2, 6.6, 7.0]
fig = plt.figure(figsize=(12, 6))
plt.subplot(1, 1, 1)
plt.plot(x, y, color='r', linestyle='-')
plt.show()
Python3快速入门（十六）——Matplotlib绘图
4、柱状图
# -*- coding=utf-8 -*-
import matplotlib.pyplot as plt
import numpy as np
if __name__ == "__main__":
x = [1, 2, 3, 4, 5]
y = [2.3, 3.4, 1.2, 6.6, 7.0]
plt.figure()
plt.bar(x, y)
plt.title("bar")
plt.show()
Python3快速入门（十六）——Matplotlib绘图
5、饼状图
# -*- coding=utf-8 -*-
import matplotlib.pyplot as plt
import numpy as np
if __name__ == "__main__":
y = [2.3, 3.4, 1.2, 6.6, 7.0]
plt.figure()
plt.pie(y)
plt.title('PIE')
plt.show()
Python3快速入门（十六）——Matplotlib绘图
6、散点图
# -*- coding=utf-8 -*-
import matplotlib.pyplot as plt
import numpy as np
if __name__ == "__main__":
n = 1024
X = np.random.normal(0, 1, n)
Y = np.random.normal(0, 1, n)
T = np.arctan2(Y, X)
plt.axes([0.025, 0.025, 0.95, 0.95])
plt.scatter(X, Y, s=75, c=T, alpha=.5)
plt.xlim(-1.5, 1.5), plt.xticks([])
plt.ylim(-1.5, 1.5), plt.yticks([])
plt.show()
Python3快速入门（十六）——Matplotlib绘图
7、等高线图
# -*- coding=utf-8 -*-
import matplotlib.pyplot as plt
import numpy as np
def get_height(x, y):
# the height function
return (1 - x / 2 + x ** 5 + y ** 3) * np.exp(-x ** 2 - y ** 2)
if __name__ == "__main__":
n = 256
x = np.linspace(-3, 3, n)
y = np.linspace(-3, 3, n)
X, Y = np.meshgrid(x, y)
plt.figure(figsize=(14, 8))
plt.contourf(X, Y, get_height(X, Y), 16, alpah=0.7, cmap=plt.cm.hot)
#
C = plt.contour(X, Y, get_height(X, Y), 16, color='black', linewidth=.5)
# adding label
plt.clabel(C, inline=True, fontsize=16)
plt.xticks(())
plt.yticks(())
plt.show()
Python3快速入门（十六）——Matplotlib绘图
8、数据3D图
# -*- coding=utf-8 -*-
import matplotlib.pyplot as plt
import numpy as np
from mpl_toolkits.mplot3d import Axes3D
if __name__ == "__main__":
fig = plt.figure()
ax = Axes3D(fig)
X = np.arange(-4, 4, 0.25)
Y = np.arange(-4, 4, 0.25)
X, Y = np.meshgrid(X, Y)
R = np.sqrt(X ** 2 + Y ** 2)
Z = np.sin(R)
ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap=plt.cm.hot)
ax.contourf(X, Y, Z, zdir='z', offset=-2, cmap=plt.cm.hot)
ax.set_zlim(-2, 2)
plt.show()
Python3快速入门（十六）——Matplotlib绘图
四、Matplotlib应用示例
1、图片加载与保存
图片加载显示：
# -*- coding=utf-8 -*-
import matplotlib.pyplot as plt
if __name__ == "__main__":
img = plt.imread('network.png')
plt.imshow(img)
plt.show()图片保存：
# -*- coding=utf-8 -*-
import matplotlib.pyplot as plt
import numpy as np
if __name__ == "__main__":
x1 = np.linspace(0, 2 * np.pi, 100)
y1 = np.sin(x1)
plt.plot(x1, y1)
x2 = x1 = np.linspace(0, 2 * np.pi, 100)
y2 = np.cos(x1)
plt.plot(x2, y2)
plt.legend(['sin(x)', 'cos(x)'], loc=0, ncol=1)
plt.savefig('test.png')
plt.show()2、多曲线实例
# -*- coding=utf-8 -*-
import matplotlib.pyplot as plt
import numpy as np
if __name__ == "__main__":
# 创建一个 8 * 6 点（point）的图，并设置分辨率为 80
plt.figure(figsize=(8, 6), dpi=80)
# 创建一个新的 1 * 1 的子图，接下来的图样绘制在其中的第 1 块（也是唯一的一块）
plt.subplot(1, 1, 1)
X = np.linspace(-np.pi, np.pi, 256, endpoint=True)
C, S = np.cos(X), np.sin(X)
# 绘制余弦曲线，使用蓝色的、连续的、宽度为 1 （像素）的线条
plt.plot(X, C, color="blue", linewidth=2.5, linestyle="-")
# 绘制正弦曲线，使用绿色的、连续的、宽度为 1 （像素）的线条
plt.plot(X, S, color="red", linewidth=2.5, linestyle="-")
# 坐标轴的范围
xmin, xmax = X.min(), X.max()
ymin, ymax = C.min(), C.max()
# 计算坐标轴的冗余
dx = (xmax - xmin) * 0.2
dy = (ymax - ymin) * 0.2
# 设置横轴的上下限
plt.xlim(xmin - dx, xmax + dx)
# 设置纵轴的上下限
plt.ylim(ymin - dy, ymax + dy)
# 设置横轴记号
plt.xticks([-np.pi, -np.pi/2, 0, np.pi/2, np.pi], [r'$-\pi$', r'$-\pi/2$', r'$0$', r'$+\pi/2$', r'$+\pi$'])
# 设置纵轴记号
plt.yticks([-1, 0, +1], [r'$-1$', r'$0$', r'$+1$'])
# 设置坐标轴位置
ax = plt.gca()
ax.spines['right'].set_color('none')
ax.spines['top'].set_color('none')
ax.xaxis.set_ticks_position('bottom')
ax.spines['bottom'].set_position(('data', 0))
ax.yaxis.set_ticks_position('left')
ax.spines['left'].set_position(('data', 0))
# 设置图例
plt.plot(X, C, color="blue", linewidth=2.5, linestyle="-", label="cosine")
plt.plot(X, S, color="red", linewidth=2.5, linestyle="-", label="sine")
plt.legend(loc='upper left')
# 在2pi/3位置做标注
t = 2 * np.pi / 3
plt.plot([t, t], [0, np.cos(t)], color='blue', linewidth=2.5, linestyle="--")
plt.scatter([t, ], [np.cos(t), ], 50, color='blue')
plt.annotate(r'$\sin(\frac{2\pi}{3})=\frac{\sqrt{3}}{2}$',
xy=(t, np.sin(t)), xycoords='data',
xytext=(+10, +30), textcoords='offset points', fontsize=16,
arrowprops=dict(arrowstyle="->", connectionstyle="arc3,rad=.2"))
plt.plot([t, t], [0, np.sin(t)], color='red', linewidth=2.5, linestyle="--")
plt.scatter([t, ], [np.sin(t), ], 50, color='red')
plt.annotate(r'$\cos(\frac{2\pi}{3})=-\frac{1}{2}$',
xy=(t, np.cos(t)), xycoords='data',
xytext=(-90, -50), textcoords='offset points', fontsize=16,
arrowprops=dict(arrowstyle="->", connectionstyle="arc3,rad=.2"))
# 坐标轴刻度标签半透明化
for label in ax.get_xticklabels() + ax.get_yticklabels():
label.set_fontsize(16)
label.set_bbox(dict(facecolor='white', edgecolor='None', alpha=0.65))
plt.show()
Python3快速入门（十六）——Matplotlib绘图
3、嵌套图实例
# -*- coding=utf-8 -*-
import matplotlib.pyplot as plt
import numpy as np
if __name__ == "__main__":
fig = plt.figure(figsize=(10, 6))
fig.set_facecolor('white')
x = [1, 2, 3, 4, 5, 6, 7]
y = [1, 3, 4, 2, 5, 8, 6]
# 大图
left, bottom, width, weight = 0.1, 0.1, 0.8, 0.8
ax = fig.add_axes([left, bottom, width, weight])
ax.plot(x, y, 'r')
ax.set_xlabel(r'$X$')
ax.set_ylabel(r'$Y$')
ax.set_title(r'$BigFigure$')
ax.spines['right'].set_color('none')
ax.spines['top'].set_color('none')
# 左上小图
left, bottom, width, weight = 0.2, 0.6, 0.25, 0.25
ax1 = fig.add_axes([left, bottom, width, weight])
ax1.plot(y, x, 'b')
ax1.set_xlabel(r'$x$')
ax1.set_ylabel(r'$y$')
ax1.set_title(r'$figure1$')
ax1.spines['right'].set_color('none')
ax1.spines['top'].set_color('none')
plt.show()
Python3快速入门（十六）——Matplotlib绘图
4、Pandas绘图
可以直接使用Pandas的Series、DataFrame实例的plot直接进行绘图。 Series示例如下：
# -*- coding=utf-8 -*-
import matplotlib.pyplot as plt
import numpy as np
import pandas as pd
if __name__ == "__main__":
# Series绘图
x = np.linspace(0, 2 * np.pi, 100)
# 正弦曲线
y = np.sin(x)
s = pd.Series(data=y, index=x)
s.plot()
plt.show()DataFrame实例：
# -*- coding=utf-8 -*-
import matplotlib.pyplot as plt
import numpy as np
import pandas as pd
if __name__ == "__main__":
# DataFrame绘图
x = np.linspace(0, 2 * np.pi, 100)
df = pd.DataFrame(data={'sin': np.sin(x), 'cos': np.cos(x)}, index=x)
df.plot()
# 取出某列数据进行绘图
# df['sin'].plot()
plt.show()DataFrame绘制柱状图：
# -*- coding=utf-8 -*-
import matplotlib.pyplot as plt
import numpy as np
import pandas as pd
if __name__ == "__main__":
df = pd.DataFrame(np.random.randint(0, 10, size=(8, 4)), index=list('abcdefgh'), columns=list('ABCD'))
ax = df.plot(kind='bar')
ax.spines['right'].set_color('none')
ax.spines['top'].set_color('none')
plt.show()kind='barh'参数表示绘制水平柱状图。
本文参与 腾讯云自媒体分享计划，分享自作者个人站点/博客。
原始发表：2019-09-27 ，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读
目录