如何使用相同的功能绘制和保存多个图表或图形?

Question

我试图使用相同的函数保存两个地块:

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

然而，第二个输出覆盖了第一个输出，我怎么能改变这一点，给我两个数字保存。无花果吗?

示例代码如下:

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

这是因为每次调用函数都使用相同的图形保存位置。您可以使用另一个参数来保存图形。这样的:

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)