import matplotlib.pyplot as plt
import matplotlib as mpl
import numpy as np
fig, ax = plt.subplots() #创建一个包含一个axes的figure
ax.plot([5, 6, 7, 8], [4, 8, 0, 10]) # 绘制图像
和MATLAB命令类似, 你还可以通过一种更简单的方式绘制图像, matplotlib.pyplot方法能够直接在当前axes上绘制图像, 如果用户为指定axes, matplotlib会帮你自己创建一个, 所以上面的例子也可以简化为下面的代码
plt.plot([5, 6, 7, 8], [4, 8, 0, 10])
x = np.linspace(0, 3, 10)
fig, ax = plt.subplots()
ax.plot(x, x, label="linear")
ax.plot(x, x**2, label="quadratic")
ax.plot(x, x**3, label="cubic")
ax.set_xlabel("x label")
ax.set_ylabel("y label")
ax.set_title("Simple Plot")
ax.legend()
第二种、采用自动绘图
x = np.linspace(0, 1.5, 10)
plt.plot(x, x, label="linear")
plt.plot(x, x**2, label="quadratic")
plt.plot(x, x**3, label ="cubic")
plt.xlabel("x label")
plt.ylabel("y label")
plt.title("Simple Plot")
plt.legend() # legend模块定义了legend类,负责绘制与轴和/或图形相关的图例。
# step1 准备数据
x = np.linspace(0, 2, 100)
y = x**2
#step2 设置绘图样式, 这一模块的扩展参考第五章进一步学习, 这一步不是必须的, 样式也可以再绘制
mpl.rc("lines", linewidth = 4, linestyle="-.")
#step3 定义布局,
fig, ax = plt.subplots()
#step4 绘制图像
ax.plot(x, y, label="linear")
#step5 添加标签, 文字和图例,
ax.set_xlabel("x label")
ax.set_ylabel("y label")
ax.set_title("Simple Plot")
ax.legend()
使用依赖pyplot自动创建figure和axes, 并绘制图像
x = np.linspace(0, 2, 100)
y = x**2
plt.plot(x, y, label="linear")
plt.plot(x, x*2, label="quadratic")
plt.xlabel("x label")
plt.ylabel("y label")
plt.title("Simple Plot")
plt.legend()