我一直在从实验中收集数据并将它们转储到.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"
。如果我尝试放置labels='files'
,则所有图都标记为无用的文件。有谁知道如何根据变量为一个情节分配一个标签?
答案 0 :(得分:2)
您想使用label
,而不是lables
或labels
。
plot(temp,label = files)