hele-mappen-colormapped-hist.py

# Importerer alle de libs der skal bruges og maaske lidt mere
from pandas import DataFrame, read_csv
from matplotlib import cm
from matplotlib.font_manager import FontProperties
import matplotlib.pyplot as plt
import pandas as pd
import sys
import matplotlib
import glob
import os
import random

# Skriver lidt ting ud om versioner
print('Python version ' + sys.version)
print('Pandas version ' + pd.__version__)
print('Matplotlib version ' + matplotlib.__version__)

# Funktion, som kan generere en vaerdi mellem 0 og 255 til random farver
r = lambda: random.randint(0,255)
cmap = cm.get_cmap('jet')

# Saetter stoerrelsen paa teksten ned
fontP = FontProperties()
fontP.set_size('small')

# Finder alle filerne
path = './'
files = []
all_files = glob.glob(path + "*.csv")

# Fjerner _resultater.csv, hvis den er der
try:
	all_files.remove('./_resultater.csv')
except:
	pass

for file in all_files:
	vars = file.split("+")
	df = pd.read_csv(file, names=['data'], header=None)['data']
	ax = df.hist(bins=100)
	fig = ax.get_figure()
	fig.savefig(file.replace(".csv",'-hist.png'))
	fig.clf()

# Loeber alle filer igennem og laver diagrammer af dem
for file in all_files:
	vars = file.split("+")
	vars = vars[1].split(" -")
	print vars[0].replace(".csv","")
	files.append([file,vars[0].replace(".csv","")])
	df = pd.read_csv(file, names=[vars[0].replace(".csv","")], header=None)
	ax = df.plot()
	fig = ax.get_figure()
	fig.savefig(file.replace(".csv",'.png'))
	fig.clf()

# Laver en liste med alle de fundne kurver
ax = None
frame = pd.DataFrame()
list_ = []
for file in files:
	df = pd.read_csv(file[0], names=[file[1]], header=None, index_col=None)
	list_.append(df)

# Gemmer den samlede fil
frame = pd.concat(list_)
ax = frame.plot(cmap=cmap)
ax.set_xlabel('Sample')
ax.set_ylabel('Latency [ms]')
box = ax.get_position()
ax.set_position([box.x0, box.y0+box.height*0.25, box.width, box.height*0.75])
ax.legend(bbox_to_anchor=(0.5, -0.05), loc='upper center', ncol=5, fancybox=True, shadow=True, prop=fontP)
fig = ax.get_figure()
fig.savefig('foo.png')