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'