analistica/ex-6/dist-plot.py

37 lines
942 B
Python
Executable File

#!/usr/bin/env python
from pylab import *
import sys
rcParams['font.size'] = 12
n = int(input())
a, b, f = loadtxt(sys.stdin, unpack=True)
subplot(131)
title('FFT')
xlabel('EDM')
ylabel('counts')
hist(a[:n], insert(b[:n], 0, a[0]), weights=f[:n],
histtype='stepfilled', color='#e3c5ca', edgecolor='#92182b')
ticklabel_format(style='sci', axis='x', scilimits=(0, 0), useMathText=True)
subplot(132)
title('RL')
xlabel('EMD')
ylabel('counts')
hist(a[n:2*n], insert(b[n:2*n], 0, a[n]), weights=f[n:2*n],
histtype='stepfilled', color='#e3c5ca', edgecolor='#92182b')
ticklabel_format(style='sci', axis='x', scilimits=(0, 0), useMathText=True)
subplot(133)
title('convolution')
xlabel('EMD')
ylabel('counts')
hist(a[2*n:], insert(b[2*n:], 0, a[2*n]), weights=f[2*n:],
histtype='stepfilled', color='#e3c5ca', edgecolor='#92182b')
ticklabel_format(style='sci', axis='x', scilimits=(0, 0), useMathText=True)
tight_layout()
show()