具有功能matplotlib的多个图

Question

我创建了一个创建绘图的函数，基本上该函数如下所示：

def draw_line(array):
    fig, ax = plt.subplots()
    ax.plot(array)

我想知道是否想在一个图形中进行多个绘图时调用此函数。 特别是，我想做类似的事情：

fig, axes = plt.subplots(nrows=2, ncols=3)
for i in list:
axes[i] = draw_line(*list[i]) 

但是，我得到的是一个带有下面实际图的空网格。

Answer 1

您不想每次调用draw_line（）时都调用新的plt.subplots（）。 而是要使用现有的轴对象。 在这种情况下，您想传递每个子图的轴及其相应的数据。 然后将两者绘制在一起。

from matplotlib import pyplot as plt
import numpy as np

def draw_line(ax,array):
    # fig, ax = plt.subplots()
    ax.plot(array)

# example data and figure
example_list = [[1,2,3],[4,5,6],[3,2,5],[3,2,5],[3,2,5],[3,2,5]]
fig, axes = plt.subplots(nrows=2, ncols=3)

# loop over elements in subplot and data, plot each one
for ax,i in zip(axes.flatten(),example_list):
    draw_line(ax,i) 

输出看起来像这样

Answer 2

@ user2241910的替代者，

from matplotlib import pyplot as plt

fig = plt.figure()
example_list = [[1,2,3],[4,5,6],[3,2,5],[5,2,3],[1,3,1],[5,3,5]]

for i,data in enumerate(example_list):
    ax = plt.subplot(2,3,i+1)
    ax.plot(data)

产生：