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'