ex-1: change name from slides.py to misc.py

Also add the possibility to create the median.pdf plot.
This commit is contained in:
Giù Marcer 2020-06-10 16:18:56 +02:00 committed by rnhmjoj
parent 336e8aa734
commit bbc1945609
2 changed files with 15 additions and 2 deletions

View File

@ -11,11 +11,12 @@ import sys
#
# -n → agrs.note → do the Landau plot with the notes
# -b → args.both → do the Landau-Moyal plot
# -m → args.area → do the Landau median plot
def moyal(x, μ, σ):
N = (1)/(np.sqrt(2 * np.pi) * σ)
return N * np.exp(- 0.5 * ((x - μ)/σ + np.exp( - (x - μ)/σ)))
return N * np.exp(- 0.5 * ((x - μ)/σ + np.exp(- (x - μ)/σ)))
def main(args):
@ -25,7 +26,7 @@ def main(args):
elif args.save:
plt.rcParams['font.size'] = 8
if args.both:
if (args.both or args.area):
plt.figure(figsize=(3, 2))
elif args.note:
plt.figure(figsize=(5, 3))
@ -69,6 +70,13 @@ def main(args):
x, y = np.loadtxt(sys.stdin, unpack=True)
plt.plot(x, y, color='#92182b', label='Landau')
# Color the area under the function
if args.area:
median = 1.3557804
x0 = np.argmin(abs(median - x))
plt.fill_between(x[:x0], y[:x0], color='#ff99a8')
plt.fill_between(x[x0:], y[x0:], color='#b28e94')
# do Moyal plot
if args.both:
μ = -0.22278298
@ -86,6 +94,8 @@ def main(args):
plt.savefig('notes/images/1-notes.pdf')
if args.both:
plt.savefig('slides/images/both-pdf.pdf', transparent=True)
if args.area:
plt.savefig('slides/images/median.pdf', transparent=True)
if __name__ == '__main__':
@ -103,5 +113,8 @@ if __name__ == '__main__':
parser.add_argument('-n', '--note',
action='store_true', default=False,
help='do the note plot')
parser.add_argument('-a', '--area',
action='store_true', default=False,
help='do the area plot')
args = parser.parse_args()
main(args)

BIN
slides/images/median.pdf Normal file

Binary file not shown.