如何使用同一功能绘制和保存多个图表或图形？

Question

我正在尝试使用相同的功能保存两个绘图：

``` plot_xy(x,y,x1,y1)``` 

``` plot_xy(x=x,y=y,x1=x2,y1=y2)``` 

但是，第二个输出覆盖了第一个输出，我如何更改此设置，以使用save.fig提供两个图形

示例代码如下所示：

import numpy as np 
import matplotlib.pyplot as plt



def plot_xy(x,y,x1,y1,*args, **kwargs):
    A = np.amax(x1)
    fig, ax = plt.subplots()
    ax.plot(x, y, 'r-', label='a' )
    ax.plot(x1, y1, linestyle='none', marker='^', label = 'b')
    ax.set(xlabel='cm', ylabel='(d/t)',
           title='ABC')
    
    ax.grid()
    ax.set_ylim(ymin=0)
    ax.set_xlim(xmin=0)
    ax.set_ylim(ymax=1)
    ax.set_xlim(xmax=A) 

    plt.legend(loc = 'best')

    plt.show()
    fig.savefig(r'C:\...location....\test.png')
    
    return fig 



    
if __name__ == "__main__": 
    """passing a rough estimated data"""
    print (0)    
    
    x = np.linspace(0,500, num=20 )
    y = np.linspace(0,0.85, num=20 )  
    
    x1 = np.linspace(0,1000, num=20 )
    y1 = np.linspace(0,1000, num=20 )
    
    x2 = np.linspace(0,2000, num=20 )
    y2 = np.linspace(0,1, num=20 )
    
    plot_xy(x,y,x1,y1)
    
    plot_xy(x=x,y=y,x1=x2,y1=y2)

谢谢你，非常感谢你的努力

Answer 1

这是因为file name是相同的。因此，上一个文件将被覆盖。您可以通过创建新文件名来处理此问题。您可以使用current seconds来确保它是不同的。或者您可以使用任何其他类型，以便每次生成的文件名都不同

import numpy as np 
import matplotlib.pyplot as plt
import datetime



def plot_xy(x,y,x1,y1,*args, **kwargs):
    A = np.amax(x1)
    fig, ax = plt.subplots()
    ax.plot(x, y, 'r-', label='a' )
    ax.plot(x1, y1, linestyle='none', marker='^', label = 'b')
    ax.set(xlabel='cm', ylabel='(d/t)',
           title='ABC')
    
    ax.grid()
    ax.set_ylim(ymin=0)
    ax.set_xlim(xmin=0)
    ax.set_ylim(ymax=1)
    ax.set_xlim(xmax=A) 

    plt.legend(loc = 'best')

    plt.show()
    fig.savefig(r'C:\...location....\test_' + str(datetime.datetime.now().strftime('%f'))+'.png')
    
    return fig 



    
if __name__ == "__main__": 
    """passing a rough estimated data"""
    print (0)    
    
    x = np.linspace(0,500, num=20 )
    y = np.linspace(0,0.85, num=20 )  
    
    x1 = np.linspace(0,1000, num=20 )
    y1 = np.linspace(0,1000, num=20 )
    
    x2 = np.linspace(0,2000, num=20 )
    y2 = np.linspace(0,1, num=20 )
    
    plot_xy(x,y,x1,y1)
    
    plot_xy(x=x,y=y,x1=x2,y1=y2)

Answer 2

这是因为对函数的每次调用都使用相同的figure save位置。您可以使用另一个参数来保存图形。如下所示：

import numpy as np 
import matplotlib.pyplot as plt



def plot_xy(filename,x,y,x1,y1,*args, **kwargs):
    A = np.amax(x1)
    fig, ax = plt.subplots()
    ax.plot(x, y, 'r-', label='a' )
    ax.plot(x1, y1, linestyle='none', marker='^', label = 'b')
    ax.set(xlabel='cm', ylabel='(d/t)',
           title='ABC')
    
    ax.grid()
    ax.set_ylim(ymin=0)
    ax.set_xlim(xmin=0)
    ax.set_ylim(ymax=1)
    ax.set_xlim(xmax=A) 

    plt.legend(loc = 'best')

    plt.show()
    
    fig.savefig(f"{filename}.png")



    
if __name__ == "__main__": 
    """passing a rough estimated data"""
    print (0)    
    
    x = np.linspace(0,500, num=20 )
    y = np.linspace(0,0.85, num=20 )  
    
    x1 = np.linspace(0,1000, num=20 )
    y1 = np.linspace(0,1000, num=20 )
    
    x2 = np.linspace(0,2000, num=20 )
    y2 = np.linspace(0,1, num=20 )
    
    plot_xy("image1",x,y,x1,y1)
    
    plot_xy("image2",x=x,y=y,x1=x2,y1=y2)