caffe loss 曲线（python 正则表达式）

import matplotlib.pyplot as plt
import re
##读取文件
fp = open(r'C:\Users\yangwenbin\Desktop\abc.log', 'r')
iterations = []  #初始化存放迭代次数的列表
loss = []  #初始化存放loss的列表
train_loss=[]  #初始化存放train_loss的列表

for ln in fp:  #按行读取文件中的子符
    if '] Iteration ' in ln and 'loss = ' in ln:

        arr1 = re.findall(r'ion (\d+)\b',ln)   #从每行的 ion开始读区数字+d 空格结束
        arr1=''.join(arr1)  #将列表转为字符
        arr2=re.findall(r'loss = (\d+)\b',ln) #从每行的loss = 开始查找字符 遇到空格时结束 
        arr2=''.join(arr2)
        #print(arr2)
        arr2=float(arr2)/10000.0
        #print(arr2)
        #print(arr2)
        loss.append(arr2)  #将字符添加至列表中
        iterations.append(arr1)
        #print(iterations)  #打印
        #print(loss)

    if '#0:' in ln and 'loss' in ln:
        arr3=re.findall(r'loss = (\d+)\b',ln)
        arr3=''.join(arr3) 
        arr3=float(arr3)/10000.0
        train_loss.append(arr3)
        #print(train_loss)
fp.close() #关闭文件夹


#plt.plot(iterations,train_loss)
plt.plot(iterations,loss)
plt.title("Iteration-Loss")
plt.xlabel("Iterations")
plt.ylabel("Loss")
plt.show()


[BEGIN] 2017/4/23 20:46:12
I0423 20:44:27.897979 27010 solver.cpp:242] Iteration 0 (0 iter/s, 3.3676s/20 iter), loss = 31519.8
I0423 20:44:27.898056 27010 solver.cpp:261]     Train net output #0: loss = 31519.8 (* 1 = 31519.8 loss)
I0423 20:44:27.898078 27010 sgd_solver.cpp:106] Iteration 0, lr = 1e-14
I0423 20:46:26.609866 27010 solver.cpp:242] Iteration 20 (0.16847 iter/s, 118.715s/20 iter), loss = 26665
I0423 20:46:26.609956 27010 solver.cpp:261]     Train net output #0: loss = 39377.8 (* 1 = 39377.8 loss)
I0423 20:46:26.609982 27010 sgd_solver.cpp:106] Iteration 20, lr = 1e-14
I0423 20:48:27.291333 27010 solver.cpp:242] Iteration 40 (0.165721 iter/s, 120.685s/20 iter), loss = 21854.6
I0423 20:48:27.291419 27010 solver.cpp:261]     Train net output #0: loss = 7725.07 (* 1 = 7725.07 loss)
I0423 20:48:27.291445 27010 sgd_solver.cpp:106] Iteration 40, lr = 1e-14
I0423 20:50:19.004346 27010 solver.cpp:242] Iteration 60 (0.179025 iter/s, 111.716s/20 iter), loss = 26581.3
I0423 20:50:19.004436 27010 solver.cpp:261]     Train net output #0: loss = 21080.2 (* 1 = 21080.2 loss)
I0423 20:50:19.004461 27010 sgd_solver.cpp:106] Iteration 60, lr = 1e-14
I0423 20:52:26.421792 27010 solver.cpp:242] Iteration 80 (0.15696 iter/s, 127.421s/20 iter), loss = 24122.7
I0423 20:52:26.421875 27010 solver.cpp:261]     Train net output #0: loss = 51531.7 (* 1 = 51531.7 loss)
I0423 20:52:26.421901 27010 sgd_solver.cpp:106] Iteration 80, lr = 1e-14
I0423 20:54:27.741914 27010 solver.cpp:242] Iteration 100 (0.164848 iter/s, 121.324s/20 iter), loss = 19128.5
I0423 20:54:27.741997 27010 solver.cpp:261]     Train net output #0: loss = 26323.5 (* 1 = 26323.5 loss)
I0423 20:54:27.742018 27010 sgd_solver.cpp:106] Iteration 100, lr = 1e-14
I0423 20:56:36.750946 27010 solver.cpp:242] Iteration 120 (0.155024 iter/s, 129.013s/20 iter), loss = 24198.6
I0423 20:56:36.751029 27010 solver.cpp:261]     Train net output #0: loss = 15857.1 (* 1 = 15857.1 loss)
I0423 20:56:36.751055 27010 sgd_solver.cpp:106] Iteration 120, lr = 1e-14
I0423 20:58:35.243075 27010 solver.cpp:242] Iteration 140 (0.168783 iter/s, 118.495s/20 iter), loss