使用subplot绘制多个子图
import matplotlib.pyplot as plt
import numpy as np
# 创建一个figure
plt.figure()
# 设置为2 * 2 的图 1 为(1,1)的子图
plt.subplot(2, 2, 1)
a = np.arange(512)
b = a**2
plt.plot(a, b)
plt.xticks([-1, -0.75, -0.5, -0.25, 0, 0.25, 0.5, 0.75, 1])
# 设置为2 * 2 的图 1 为(1,2)的子图
plt.subplot(2,2,2)
n = 1024
X = np.random.normal(0, 1, n)
Y = np.random.normal(0, 1, n)
T = np.arctan2(Y, X) # for color value
plt.scatter(X, Y, c=T, alpha=0.5)
plt.xlim((-1.5, 1.5))
plt.ylim((-1.5, 1.5))
# 设置为2 * 2 的图 1 为(2,1)的子图
plt.subplot(2,2,3)
def f(x, y):
#The height function
return (1 - x/2 + x**5 + y**3) * np.exp(-x**2-y**2)
n = 256
x = np.linspace(-3, 3, n)
y = np.linspace(-3, 3, n)
X, Y = np.meshgrid(x, y)
# Use plt.contourf to filling contours 等高线
# X, Y and value for (X,Y) point
plt.contourf(X, Y, f(X, Y), 8, alpha=0.75, cmap= plt