python matplotlib保存图

Question

它一定很容易，但我仍然无法弄清楚。假设我正在阅读很多带有glob模块的txt文件，并且做了一些处理，然后用matplotlib绘制它们。

import glob 
ascii = sorted(glob.glob('C:/Users/ENAMUL/PYTHON/*.txt')) 
for count,i in enumerate(ascii): 
........ 
........ 

现在我想保存这些数字。我可以这样做，通过计算数字来节省时间。

plt.savefig(str(count)+'png') 

但是，如果我想通过输入文件名来保存它们，我该怎么做？请任何帮助。

Answer 1

在循环中，i包含文件的文件名，所以：

import os.path 

.... 

plt.savefig(os.path.splitext(os.path.basename(i))[0] + '.png') 

它的工作原理是这样的。 os.path.basename返回文件名：

In [2]: os.path.basename('foo/bar/baz.bat') 
Out[2]: u'baz.bat' 

然后splitext确实明显：

In [3]: os.path.splitext(os.path.basename('foo/bar/baz.bat')) 
Out[3]: (u'baz', u'.bat') 

所以：

In [4]: os.path.splitext(os.path.basename('foo/bar/baz.bat'))[0] + '.png' 
Out[4]: u'baz.png' 

如果你想保留的路径，只是删除了basename呼叫，只使用splitext：

In [5]: os.path.splitext('foo/bar/baz.bat')[0] + '.png' 
Out[5]: u'foo/bar/baz.png' 

Answer 2

您在ascii存储文件名 - 所以你应该能够保存这个数字的时候使用它：

plt.savefig(ascii[count] + '.png')