/
model_A.py
44 lines (26 loc) · 958 Bytes
/
model_A.py
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
import numpy as np
import scipy.special as sps
import matplotlib.pyplot as plt
def kappa(r, z, tau=0.7, zd=0.29, hd=8.1):
kappa0 = tau/(2 * zd)
return kappa0 * np.exp(-1 * (r/hd + z/zd))
def A(r, z, tau0=0.7, zd=0.29, hd=8.1):
r = np.abs(r)
kappa0 = tau0/(2 * zd)
return 1.086 * 2 * kappa0 * r * sps.kn(1,r/hd) * np.exp(-1 * z/zd)
def A_vec(r, Z, tau0=0.2, zd=0.25, hd=7.68):
output = np.zeros(Z.size)
for i, z in enumerate(Z):
output[i] = A(r,z,tau0=tau0, zd=zd, hd=hd)
return output
def plot_A(r, tau0=0.85, zd=0.29, hd=8.1):
zlist = np.linspace(0,2.5,20)
#Alist = np.zeros(20)
# for i, z in enumerate(zlist):
# Alist[i] = A(r,z,tau0=tau0, zd=zd, hd=hd)
Alist = A_vec(r,zlist,tau0=tau0, zd=zd, hd=hd)
ax = plt.figure().add_subplot(111)
ax.set_xlabel('z [kpc]')
ax.set_ylabel('A$_V$')
ax.plot(zlist,Alist)
ax.figure.show()