Python数据分析三大框架之matplotlib（八）subplot绘制多个子图

使用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