from data2 import trials
import matplotlib.pyplot as plt

vecs = []
names = []
for trial in trials:
    vecs.append(trial.nearest_neighbor())
    names.append(str(trial))
    
plt.boxplot(vecs, labels=names)
plt.title('Nearest Neighbor for Each Pit in Different Trials')
plt.ylabel('Distance to Nearest Neighbor (cm)')
plt.savefig('boxplot.png', bbox_inches='tight')