matplotlib画子图（subplot）

Notes

• 用plt.subplots在一幅大 figure 里画几幅小图；
• 一幅小图里可能画多条曲线；
• plot画连续曲线或散点，指定线型、颜色；
• scatter散点图；
• bar柱状图，设定纵轴坐标上、下限，打数值标签；

Rendering

Code

import matplotlib.pyplot as plt
import numpy as np
from scipy.special import expit  # sigmoid

# 子图分在两行、两列
fig, ax = plt.subplots(2, 2)

# (0, 0) 左上
t = np.arange(0.0, 2.0, 0.1)
data_1 = np.sin(t * np.pi)
data_2 = np.cos(t * np.pi)
# plot 画连续曲线或散点
ax[0][0].plot(t, data_1, 'r*')
ax[0][0].plot(t, data_2, 'b-')

# (0, 1) 右上
data_3 = -7 * t - 2
ax[0][1].scatter(t, data_3)  # scatter 画散点图

# (1, 0) 左下
name_4 = ['a', 'b', 'c', 'd']
data_4 = [0.71, 0.83, 0.94, 0.68]
ax[1][0].bar(name_4, data_4, width=0.35)  # bar 画柱状图
ax[1][0].set_ylim([0, 1])  # 设定 y 轴坐标在 [0, 1] 内
for x, y in enumerate(data_4):
    ax[1][0].text(x, y, y)  # text 打数值标签

# (1, 1) 右下
data_5 = expit(t * np.pi)
ax[1][1].plot(t, data_5, 'y*')

plt.show()

Reference

1. Matplotlib的子图subplot的使用
2. matplotlib subplot 子图
3. Plotting categorical variables
4. How to set xlim and ylim for a subplot in matplotlib [duplicate]
5. Grouped bar chart with labels
6. matplotlib可视化之如何给图形添加数据标签？
7. matplotlib模块数据可视化-绘制柱状图
8. Adding value labels on a matplotlib bar chart