欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

matplotlib的subplot使用

程序员文章站 2022-03-21 21:12:50
...

官方docs:https://matplotlib.org/api/_as_gen/matplotlib.pyplot.subplots.html?highlight=subplots#matplotlib.pyplot.subplots

matplotlib.pyplot.subplots

Create a figure and a set of subplots

调用subplots方法会返回一个figure和一个subplot的set,使用方法比较简单

例子:创建4x4的subplot,没个subplot都显示一个矩阵(使用imshow和matshow都可以,本例是gray image)

代码来自:https://github.com/matplotlib/matplotlib/issues/4976

import numpy as np
import matplotlib.pyplot as plt

fig, axes = plt.subplots(4, 4)

for i, ax in enumerate(axes.ravel()):
    im = ax.imshow(np.random.normal(size=100).reshape([10,10]))
    ax.set_title(i)

plt.tight_layout()

顺便:

numpy.ravel() 和 numpy.flatten()都是讲矩阵碾平的意思,大体的意思是

ravel()返回的是像指针一样的引用,赋值给新变量后,修改新变量会影响原变量

flatten()返回的是原变量的一个备份,修改新变量不会修改原变量