python: 使用matplotlib的pyplot绘制图表

#!encode=utf8
from matplotlib import pyplot as plt
import sys
ignore_num=(int)(sys.argv[1])
data=sys.argv[2]
show_type=0
if len(sys.argv)>3:
    show_type=(int)(sys.argv[3])
x=[]
valid_ppl=[]
train_ll=[]
valid_ll=[]
fi=open(data,"r")
line=fi.readline()
n=0
while line:
    if line.find("Values") >= 0:
        arr=line.split(" ")
        n+=1
        if n>ignore_num:
            x.append(n)
            train_ll.append(arr[6])
            valid_ll.append(arr[8])
            valid_ppl.append(arr[9])
    line=fi.readline()
if show_type==0:
    plt.plot(x,train_ll,color="blue",label="train_ll")
    plt.plot(x,valid_ll,color="red",label="valid_ll")
    plt.plot(x,valid_ppl,color="orange",label="valid_ppl")
elif show_type==1:
    plt.plot(x,train_ll,color="blue",label="train_ll")
    plt.plot(x,valid_ll,color="red",label="valid_ll")
else:
    plt.plot(x,valid_ppl,color="orange",label="valid_ppl")
plt.legend(loc=‘upper right‘)
plt.show()