summaryrefslogtreecommitdiff
path: root/project/scripts/cos_pi_x.py
blob: 551c142f0cf90c2f01c008b47c602fea91e28483 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
#!/usr/bin/python


import numpy as np
import matplotlib.pyplot as plt
from matplotlib.patches import Polygon


plt.rcParams['legend.fontsize'] = 10

fig = plt.figure ()
ax = fig.gca ()

x = np.linspace (-2, 2, 5000)
y = np.cos (np.pi * x)

ax.plot (x, y, label='$y=Re((-1)^{x})$')
ax.legend ()

a, b = 0, 1
ix = np.linspace (a, b)
iy = np.cos (np.pi * ix)
verts = [(a,0)] + zip (ix, iy) + [(b,0)]
poly = Polygon (verts, facecolor='0.9', edgecolor='0.5')
ax.add_patch (poly)

ax.set_xticks ([-2, -1, 1, 2])
ax.spines['left'].set_position ('center')
ax.spines['right'].set_position ('center')

ax.set_yticks ([-1, -0.5, 0.5, 1])
ax.spines['top'].set_position ('center')
ax.spines['bottom'].set_position ('center')

plt.show ()