在matplotlib，Python中绘制几个图形

Question

我需要在python matplotlib中绘制图形。

对于每个图形，我都会先进行一些计算，然后绘制图形。 这是一些代码：

import matplotlib.pyplot as plt

def draw(LuxCoordinates, finalPix, verName):
    plt.axes([0.2, 0.2, 0.7, 0.6]);
    plt.xlim(0,3500); #Axis x
    plt.ylim(0,100); #Axis y
    plt.plot(LuxCoordinates, finalPix, 'g');
    plt.scatter(LuxCoordinates, finalPix, linewidths=1)
    plt.grid(axis)
    plt.xlabel('X_Coordinates', color='r');
    plt.ylabel('Y_Coordinates', color='r');
    plt.title(verName, color='#afeeee');
    savefig(verName+'.png');
    plt.show();

我的问题是，根据我拥有的图形数量，我两次或多次调用此函数，得到的图形分别位于不同的图形中，但是我想在同一图形上绘制所有图形，以便进行比较。 我该怎么做？ 谢谢你们！

Answer 1

删除plt.show()

您的函数应该获取figure或axis作为输入参数，并在该图形上进行绘制，然后在完成所有绘制后在函数外部进行绘制，您可以调用plt.show()

这是一个最小的示例：

import matplotlib.pyplot as plt
import numpy as np

def plotter ( ax, col ):
    data = np.random.normal( size=(50, 2 ) )
    x, y = data[:, 0], data[:, 1 ]
    ax.scatter( x, y, color=col )

fig = plt.figure()
ax = fig.add_axes([0.2, 0.2, 0.7, 0.6])

plotter( ax, 'Blue' )
plotter( ax, 'Red' )
fig.show( )