forked from CompBioClasses/Python_tutorial
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Lecture_examples
72 lines (54 loc) · 1.6 KB
/
Lecture_examples
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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
import matplotlib.pyplot as plt
import numpy as np
import math
# x = np.linspace(-math.pi, math.pi, 100)
# a = 2
# b = 3
# omega = math.pi
# y = a + b * np.sin(omega * x)
# plt.title("a+bsin($\omega$x)")
# plt.xlabel("x")
# plt.ylabel("f(x)")
# plt.plot(x,y)
# plt.show()
def func(a, b, omega):
x = np.linspace(-math.pi, math.pi, 100)
y = a + b * np.sin(omega * x)
plt.title(f"a+bsin($\omega$x)")
plt.xlabel("x")
plt.ylabel("f(x)")
plt.plot(x,y)
#plt.show()
A = np.array([i for i in range(5)])
B = np.array([i for i in range(1,6)])
Omega = np.array([i for i in range(5)])
#Third bullet
def individual_plots(A_array, B_array, Omega_array):
for i in range(len(A)):
a = A_array[i]
b = B_array[i]
omega = Omega_array[i]
func(a, b, omega)
plt.show()
#Fourth bulet
def same_plot(A_array, B_array, Omega_array):
for i in range(len(A_array)):
a = A_array[i]
b = B_array[i]
omega = Omega_array[i]
func(a, b, omega)
plt.show()
#same_plot(A, B, Omega)
#Fifth bullet
import argparse
parser = argparse.ArgumentParser()
parser.add_argument("A", type=float, help="Translation.")
parser.add_argument("B", type=float, help="Amplitude dilation.")
parser.add_argument("Omega", type=float, help="Period shift parameter.")
parser.add_argument("-n", action = "store_true", help = "true: same plot ; false (default): individual plots")
if __name__ == '__main__':
args = parser.parse_args()
if args.n is True:
individual_plots(args.A, args.B, args.Omega)
else:
same_plot(args.A, args.B, args.Omega)