add format plot codes

This commit is contained in:
Mhrooz 2024-08-29 09:36:33 +02:00
parent c80cfb8cac
commit a7a6906a6d

View File

@ -5,11 +5,19 @@ import pandas as pd
import argparse
def plot(l,filename):
lenth = len(l)
threshold = [0, 10000, 20000, 30000, 40000, 50000, 60000, 70000]
labels = ['0-10k', '10k-20k,', '20k-30k', '30k-40k', '40k-50k', '50k-60k', '60k-70k']
l = [i/15625 for i in l]
l = l[:7]
datasets = filename.split('_')[-1].split('.')[0]
plt.figure(figsize=(8, 6))
plt.subplots_adjust(top=0.85)
plt.ylim(0,0.3)
plt.title('Distribution of Swap Scores in ' + datasets)
plt.bar(labels, l)
for i, v in enumerate(l):
plt.text(i, v + 0.01, str(round(v, 2)), ha='center', va='bottom')
plt.savefig(filename)
def analyse(filename):