我一直在从实验中收集数据并将它们转储到 .txt 文件中,我决定尝试编写一个快速脚本来使用 matplotlib 绘制它们。绘图工作正常,但我不知道如何根据文件名标记绘图。
from numpy import *
from pylab import *
from matplotlib import rc
import sys
import os
rc('text',usetex=True)
rc('font',**{'family':'serif','serif':['Computer Modern']})
os.chdir(".")
for files in os.listdir("."):
if files.endswith(".txt"):
f = open(files,'r')
temp = []
for l in f:
temp.append(float(l))
plot(temp,labels=files)
hold(True)
legend(loc='lower left')
hold(False)
# Save the figure in a separate file
savefig('test.png')
# Draw the plot to the screen
show()
问题似乎与plot(temp,lables=files)
. 如果我把lables=files
我得到错误TypeError: There is no line property "labels"
。如果我尝试 put labels='files'
,所有的图都被标记为无用的文件。有谁知道如何根据变量为绘图分配标签?