import os
import re
import matplotlib.pyplot as plt
from sympy.physics.control.control_plots import matplotlib
# 设置默认字体
plt.rcParams['font.family'] = 'Times New Roman'
plt.rcParams['font.size'] = 4
matplotlib.rc("font", family='Microsoft YaHei')
# 设置线条宽度
plt.rcParams['lines.linewidth'] = 0.5
# 设置坐标轴标签和标题的字体大小
plt.rcParams['axes.labelsize'] = 7
plt.rcParams['axes.titlesize'] = 7
# 设置图例字体大小
plt.rcParams['legend.fontsize'] = 3
def get_data(file_pattern,laber_name):
# 存储数字和delivery_prob值的列表
x_values = []
y_values = []
# 列出目标文件夹中的所有文件
for filename in os.listdir(folder_path):
# 如果文件名符合正则表达式规则,则打开文件并查找delivery_prob的值
if re.match(file_pattern, filename) and os.path.isfile(os.path.join(folder_path, filename)):
with open(os.path.join(folder_path, filename), 'r') as file:
for line in file:
# 假设每行的格式为“delivery_prob: 0.12345”
match = re.search(laber_name+':\s+(\d+\.\d+)', line)
if match:
delivery_prob = float(match.group(1))
# 提取文件名中的数字,并将其与delivery_prob值相关联
digit_match = re.search('\d+', filename)
if digit_match:
x_value = int(digit_match.group(0))
x_values.append(x_value)
y_values.append(delivery_prob)
return x_values,y_values
def draw(x,Y,laberY,laber,Pic_i):
plt.subplot(2, 2, Pic_i + 1)
color=["b","g","r","c","m","y","k","d"]
Oshape=[".","v","+","x","1","p","s","o","<"]
c = 0
# 定义每条线的x和y坐标
print(Y)
result = []
for i in range(len(Y)): #0 1
for j in range(len(x)):
result.append([x[j], Y[i][j]])
result = sorted(result, key=lambda x: x[0])
x1 = [item[0] for item in result]
y1 = [item[1] for item in result]
plt.plot(x1, y1, color[c], label=laber[c], marker=Oshape[c], markersize=2)
c=c+1
# 添加图例
plt.legend()
plt.xlabel('bufferSize/M')
plt.ylabel(laberY)
print(result)
result = []
# plt.show()
if __name__ == '__main__':
# 文件夹路径=
folder_path=r'C:/haimianbaobao/eports/DiffBuffer'
# 正则表达式匹配规则,用于筛选文件名
laber_name=["delivery_prob","overhead_ratio","latency_avg","hopcount_avg",] #绘制的数据——修改部分2
file_name=[
r'^Prophet.*MessageStatsReport.txt$',
r'^SprayAndWaitRouter.*MessageStatsReport.txt$',
r'^对照1.*MessageStatsReport.txt$',
r'^对照2.*MessageStatsReport.txt$',
r'^自己代码名.*MessageStatsReport.txt$',
]
laber = []
for name in file_name:
pattern = re.compile(r'^(?:\^)?(.*?)\.\*')
match = pattern.search(name)
if match:
laber.append(match.group(1))
# print(laber) #['Epidemic', 'ProphetRouter', 'MDCE', 'FirstContactRouter', 'SprayAndWaitRouter']
for PictureNum in range(4):
print(PictureNum)
Y = []
labername = laber_name[PictureNum]
for filename in file_name:
x, y = get_data(filename, labername)
Y.append(y) # [[x],[[y1],[y2],[y3]]]
draw(x,Y,labername,laber,PictureNum)
final_data = Y[-1]
print(labername)
if PictureNum==0:
for i in range(len(Y) - 1):
increase_rates = [(final_data[j] - Y[i][j]) / Y[i][j]
for j in range(len(final_data))]
average_increase_rate = sum(increase_rates) / len(increase_rates)
if average_increase_rate>0:
print(f"最后一组数据相对于第{i + 1}组数据的平均提升率为: ★ {average_increase_rate * 100}%")
else:
print(f"最后一组数据相对于第{i + 1}组数据的平均下降率为: {-average_increase_rate * 100}%")
else:
for i in range(len(Y) - 1):
increase_rates = [-(final_data[j] - Y[i][j]) / final_data[j]
for j in range(len(final_data))]
average_increase_rate = sum(increase_rates) / len(increase_rates)
if average_increase_rate>0:
print(f"最后一组数据相对于第{i + 1}组数据的平均降低率为: ★ {average_increase_rate * 100}%")
else:
print(f"最后一组数据相对于第{i + 1}组数据的平均提升率为: {-average_increase_rate * 100}%")
# laber_name 为纵坐标标签,laber为 图例说明,PictureNum 为第几个图
# 调节每个子图之间的间隔
plt.subplots_adjust(top=0.97,
bottom=0.13,
left=0.075,
right=0.985,
hspace=0.285,
wspace=0.45)
plt.savefig('结果/resultDiffbufferSize.pdf')
# 显示图像,并启用滚动
plt.show(block=False)
plt.tight_layout()
求各个指标的提升率和下降率——输出到控制台
相对于最后一组数据,把要比较的数据放到最后一个。
第一个图是投递率,越高越好
后面的,越低越好
标★