evaporation/CSV_plot.py at master · DM-Colgate/evaporation · GitHub

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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
#!/usr/bin/env python
import numpy as np
import matplotlib.pyplot as plt
from pandas import DataFrame

# set up some ploting stuff
fig = plt.figure(figsize = (9,6))
plt.style.use('fast')
pallette = plt.get_cmap('magma')
pallette1 = plt.get_cmap('viridis')


# with open('py7.dat') as file:
#     lines = file.readlines()
#     x = [float(line.split()[0]) for line in lines]
#     y = [float(line.split()[1]) for line in lines]
# plt.plot(x, y, ls = '-', linewidth = 1, label='py')


# with open('TM4_725.csv') as file:
#     lines = file.readlines()
#     x = [float(line.split(',')[0]) for line in lines]
#     y = [float(line.split(',')[1]) for line in lines]
# plt.plot(x, y, label='100 $M_\odot$', c=pallette1(2/9))

with open('TM5_400.csv') as file:
    lines = file.readlines()
    x = [float(line.split(',')[0]) for line in lines]
    y = [float(line.split(',')[1]) for line in lines]
plt.plot(x, y, label='300 $M_\odot$', c=pallette1(5/9))

with open('TM6_400.csv') as file:
    lines = file.readlines()
    x = [float(line.split(',')[0]) for line in lines]
    y = [float(line.split(',')[1]) for line in lines]
plt.plot(x, y, label='1000 $M_\odot$', c=pallette1(8/9))


# with open('star2.dat') as file:
#     lines = file.readlines()
#     x = [float(line.split()[0]) for line in lines]
#     y = [float(line.split()[1]) for line in lines]
# plt.scatter(x, y, s=1, marker=',', label='5 $M_\odot$', c=pallette1(3/10))

# with open('star3.dat') as file:
#     lines = file.readlines()
#     x = [float(line.split()[0]) for line in lines]
#     y = [float(line.split()[1]) for line in lines]
# plt.scatter(x, y, s=1, marker=',', label='15 $M_\odot$', c=pallette1(4/10))

# with open('star4.dat') as file:
#     lines = file.readlines()
#     x = [float(line.split()[0]) for line in lines]
#     y = [float(line.split()[1]) for line in lines]
# plt.scatter(x, y, s=1, marker=',', label='20 $M_\odot$', c=pallette1(5/10))

# with open('star5.dat') as file:
#     lines = file.readlines()
#     x = [float(line.split()[0]) for line in lines]
#     y = [float(line.split()[1]) for line in lines]
# plt.scatter(x, y, s=1, marker=',', label='100 $M_\odot$', c=pallette1(6/10))

# with open('star6.dat') as file:
#     lines = file.readlines()
#     x = [float(line.split()[0]) for line in lines]
#     y = [float(line.split()[1]) for line in lines]
# plt.scatter(x, y, s=1, marker=',', label='400 $M_\odot$', c=pallette1(7/10))


# with open('star7.dat') as file:
#     lines = file.readlines()
#     x = [float(line.split()[0]) for line in lines]
#     y = [float(line.split()[1]) for line in lines]
# plt.scatter(x, y, s=1, marker=',', label='1000 $M_\odot$', c=pallette1(8/10))


plt.xlabel('$m_\chi$ [GeV]')
plt.ylabel('$T_{\chi}$ [K]')
# plt.title('')
plt.legend()
plt.xscale('log')
# plt.yscale('log')
plt.xlim(10**(-5), 10**5)
# plt.ylim(5*10**7, 1.5*10**8)

# fig = plt.figure()
plt.tight_layout()
plt.savefig("Tchi.pdf")
# plt.show()