【Matplotlib】在Jupyter交互页面中绘制折线图对比（自用函数）
0x00 前言
最近数据对比的任务比较常见，比如好些模型的横向对比，
对于 Loss、PRF、Hits 之类的数据，有时需要作log，有时需要去除前面几个值，
还要考虑数据不对齐、记录文件格式不一致等诸多问题，总之主需求是鲁棒性，
稍微写了个画表格的函数，暂时还比较乱，后续有时间再作优化好了，
现在暂时写在这作为记录，便于易于在不同机器上获取使用以及方便后续优化更新。
0x01 用法
# single
draw(y, y_label='Data', title='undefined title', semilog=False)
# multiple
drawx(y_list, label_list=None, title='undefined title',
start=0, end=None, x_list=None, x_label=None, semilog=False, axh=0)
0x02 Source Code
import matplotlib.pyplot as plt
from pylab import * # 支持中文
import warnings
warnings.filterwarnings('ignore')
mpl.rcParams['font.sans-serif'] = ['SimHei']
#plt.plot(x, y, 'ro-')
#plt.plot(x, y1, 'bo-')
#pl.xlim(-1, 11) # 限定横轴的范围
#pl.ylim(-1, 110) # 限定纵轴的范围
def focus(filename, label='[Valid_F1_G]', pos=1, delim='\t'):
# used for logs like '[Valid_F1_G]\t0.667\t0.667\t0.667'
target = filter(
lambda x: x.startswith(label),
[line for line in open(filename, 'r')]
)
return map(
lambda x: x.split(delim)[pos].strip(),
target
)
def draw(y, y_label='Data', title='undefined title', semilog=False):
x = range(1, y.__len__()+1)
plt.plot(x, y, marker='o', mec='r', mfc='w', label=y_label)
plt.legend() # 让图例生效
#plt.xticks(x, x, rotation=0)
plt.margins(0.05)
plt.subplots_adjust(bottom=0.15)
plt.xlabel(u"Epoch") # X轴标签
plt.title("Brief Figure for {}".format(title)) #标题
if semilog: plt.semilogy()
plt.show()
def draw2(y1, y2,
y1_label='Data1', y2_label='Data2',
semilog=False, title='undefined title', start=0, end=9999, axh=0):
length = min(end, max(y1.__len__(), y2.__len__()))
y1_len = min(length, y1.__len__())
y2_len = min(length, y2.__len__())
x = range(1, length+1)
plt.figure(figsize=(15, 5))
plt.plot(x[start:y1_len], y1[start:y1_len], marker='o', mec='r', mfc='w', label=y1_label)
plt.plot(x[start:y2_len], y2[start:y2_len], marker='X', mfc='w', ms=8, label=y2_label)
plt.legend() # 让图例生效
# plt.xticks(x1, x1, rotation=0)
plt.margins(0.05)
plt.subplots_adjust(bottom=0.15)
plt.xlabel(u"Epoch") #X轴标签
plt.title("Brief Figure for {}".format(title)) #标题
if axh: plt.axhline(axh)
if semilog: plt.semilogy()
plt.show()
def draw4(y1, y2, y3, y4,
y1_label, y2_label, y3_label, y4_label,
title = 'title', start=0, semilog=False, axh=0):
length = min(y1.__len__(), y2.__len__(), y3.__len__(), y4.__len__())
x = range(1, length+1)
plt.figure(figsize=(15, 5))
plt.plot(x[start:], y1[start:length], marker='o', mec='r', mfc='w', label=y1_label)
plt.plot(x[start:], y2[start:length], marker='X', mfc='w', ms=8, label=y2_label)
plt.plot(x[start:], y3[start:length], marker='*', mfc='w', mec='b', label=y3_label)
plt.plot(x[start:], y4[start:length], marker='.', mfc='w', label=y4_label)
plt.legend() # 让图例生效
plt.margins(0.05)
plt.subplots_adjust(bottom=0.15)
plt.xlabel(u"Epoch") #X轴标签
plt.title("Brief Figure for {}".format(title)) #标题
if axh: plt.axhline(axh)
if semilog: plt.semilogy()
plt.grid(False)
plt.show()
def drawx(y_list, label_list=None, title='undefined title',
start=0, end=None, x_list=None, x_label=None, semilog=False, axh=0):
length = min(end, max(map(len, y_list))) if end else max(map(len, y_list))
y_length = map(lambda x: min(length, x.__len__()), y_list)
x_list = range(1, length+1) if x_list is None else x_list[:length]
plt.figure(figsize=(15, 5))
if label_list is None:
label_list = map(
lambda x: u"Data_{}".format(x),
range(y_list.__len__()))
for idx, label in enumerate(label_list):
plt.plot(
x_list[start: y_length[idx]],
y_list[idx][start: y_length[idx]],
marker='o',
label=label)
plt.legend() # let legends work
plt.margins(0.05)
plt.subplots_adjust(bottom=0.15)
plt.xlabel(u"Epoch" if x_label is None else x_label)
plt.title(u"Brief Figure for {}".format(title)) #标题
if axh: # Add a horizontal line across the axis.
plt.axhline(axh)
if semilog:
plt.semilogy()
plt.grid(False)
plt.show()
```