plot.py

plot.py

import matplotlib.pyplot as plt
import csv

def pl(threads_list, token_time_list, nn):        
	if nn:
		plt.plot(threads_list, token_time_list)
	else:
		plt.plot(threads_list, token_time_list, '--', color='orange')

def plf(f, nn):
    threads_list = []
    token_time_list = []
    reader = csv.reader(f)
    for row in reader:
        threads_list.append(int(row[0]))
        token_time_list.append(float(row[1]))
    pl(threads_list, token_time_list, nn)

plt.xlabel("Number of threads")
plt.ylabel("Token time (ms)")
plt.title("Token time vs Number of threads")

with open('master.csv', newline='') as f:
    plf(f, True)

with open('threadpool.csv', newline='') as f:
    plf(f, False)

plt.show()