Das deutsche Python-Forum

Code: Alles auswählen

import numpy as np
import matplotlib as mpl
import matplotlib.pyplot as pl
import json
import itertools as it
import os
import tikzplotlib as tikz

a = {}
arr = os.listdir()
print(arr)
with open('./result_1_final/test_results.json', 'r') as f: 
    a = json.load(f)

ampl_times_fbc = [[a[j][i]['find_best_coverage']['ampl_time'] for i in range(len(a[1]))] for j in range(len(a))]
elapsed_time_fbc = [[a[j][i]['find_best_coverage']['elapsed_time'] for i in range(len(a[1]))] for j in range(len(a))]
nodes = [a[i][1]['node_number'] for i in range(len(a))]
partition_sizes = [4,5,6]
nodes_partitions = list(zip(nodes, it.cycle(partition_sizes)))


pl.style.use("ggplot")
fig1, ax1 = pl.subplots()

positions = list(range(1, len(ampl_times_fbc[0::3]) + 1))
bp = []
colors = ['red', 'green', 'blue']
for i in range(len(colors)):
    bp.append(ax1.boxplot(ampl_times_fbc[i::len(colors)], positions, showfliers=False, patch_artist=True))
    positions = [round(i, 1) for i in np.add(positions, 0.2).tolist()]

    for box in bp[i]['boxes']:
        box.set(color = colors[i], linewidth = 2)
        
tikz.save('boxplot.tex')

mpl.use('TKAgg', force = True)
pl.show()