Skip to content

Commit 18e8169

Browse files
committed
Add all changes made before thesis revision.
1 parent c9c7eae commit 18e8169

14 files changed

+12022
-172
lines changed

AccretionExample.ipynb

Lines changed: 501 additions & 0 deletions
Large diffs are not rendered by default.

complex_eigenvalues.py

Lines changed: 31 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,9 @@
1919
blocksize = 4096
2020
num_files = 12
2121
end_column = 48601
22+
#blocksize = 4096
23+
#num_files = 1
24+
#end_column = 4095
2225

2326
splits = [(blocksize*i, blocksize*(i+1)-1) for i in range(num_files)]
2427
splits[-1] = (splits[-1][0], end_column)
@@ -31,7 +34,7 @@
3134
efiles.append(nc4.Dataset(name, 'r'))
3235

3336
plt.autoscale(tight=True)
34-
cmap = plt.get_cmap('Reds')
37+
cmap = plt.get_cmap('Greys')
3538

3639
total_evalues = 0
3740
for efile in efiles:
@@ -75,7 +78,7 @@
7578
circle_rad = np.log10(1./300.)-cutoff_exp
7679
circle_x = circle_rad * np.cos(np.linspace(0., 2.*np.pi, 1001))
7780
circle_y = circle_rad * np.sin(np.linspace(0., 2.*np.pi, 1001))
78-
plt.plot(circle_x, circle_y)
81+
plt.plot(circle_x, circle_y, color='b')
7982
converge_rad = 1./300.
8083
converge_x = converge_rad * (np.cos(np.linspace(0., 2.*np.pi, 1001)) - 1.)
8184
converge_y = converge_rad * np.sin(np.linspace(0., 2.*np.pi, 1001))
@@ -84,21 +87,21 @@
8487
plot_abs = np.maximum(np.log10(converge_abs) - cutoff_exp, 0.)
8588
plot_x = plot_abs * np.cos(converge_angle)
8689
plot_y = plot_abs * np.sin(converge_angle)
87-
plt.plot(plot_x, plot_y)
90+
plt.plot(plot_x, plot_y, color='g')
8891
plt.axvline(x=0., color='k', linewidth=2.)
8992
plt.axhline(y=0., color='k', linewidth=2.)
9093
plt.axis([-max_plotval, max_plotval, -max_plotval, max_plotval])
91-
ticks = np.arange(-max_plotval, max_plotval+1., 1)
94+
ticks = np.arange(-max_plotval, max_plotval+1., 2)
9295
# The "0.01" hack below is there to ensure that "0" outputs a 10 with a positive sign.
9396
tickvals = ["${}^{{{}}}$".format(int(np.sign(i+0.01)*10),int(abs(i)+cutoff_exp)) for i in ticks]
9497
ax = plt.gca()
9598
ax.set_xticks(ticks)
96-
ax.set_xticklabels(tickvals)
99+
ax.set_xticklabels(tickvals, fontsize=16)
97100
ax.set_yticks(ticks)
98-
ax.set_yticklabels(tickvals)
99-
plt.clim(vmin=0., vmax=1.e4)
101+
ax.set_yticklabels(tickvals, fontsize=16)
102+
plt.clim(vmin=0., vmax=8.e3)
100103
plt.colorbar()
101-
plt.savefig('./complex_eigenvalues.png')
104+
plt.savefig('./complex_eigenvalues.eps')
102105
plt.close()
103106

104107
midpoint = nbins // 2
@@ -130,43 +133,51 @@
130133
plt.bar(bin_centers, real_zeros,
131134
width=(bins[1]-bins[0]), color='b', label='Real zeros')
132135
plt.legend(loc='best')
133-
ticks = np.arange(-max_plotval, max_plotval+1., 1)
136+
ticks = np.arange(-max_plotval, max_plotval+1., 2)
134137
tickvals = ["${}^{{{}}}$".format(int(np.sign(i+0.01)*10),int(abs(i)+cutoff_exp)) for i in ticks]
135138
ax = plt.gca()
139+
ax.ticklabel_format(style='sci')
136140
ax.set_xlim(left=bins[0], right=bins[-1])
137141
ax.set_xticks(ticks)
138-
ax.set_xticklabels(tickvals)
139-
plt.savefig('./complex_vertint.png')
142+
ax.set_xticklabels(tickvals, fontsize=16)
143+
plt.yticks(fontsize=16)
144+
plt.savefig('./complex_vertint.eps')
140145
plt.close()
141146

142147
plt.plot(bin_centers, real_zeros / all_zeros, color='k')
143-
ticks = np.arange(-max_plotval, max_plotval+1., 1)
148+
ticks = np.arange(-max_plotval, max_plotval+1., 2)
144149
tickvals = ["${}^{{{}}}$".format(int(np.sign(i+0.01)*10),int(abs(i)+cutoff_exp)) for i in ticks]
145150
ax = plt.gca()
151+
ax.ticklabel_format(style='sci')
146152
ax.set_xticks(ticks)
147-
ax.set_xticklabels(tickvals)
148-
plt.savefig('./complex_ratio.png')
153+
ax.set_xticklabels(tickvals, fontsize=16)
154+
plt.yticks(fontsize=16)
155+
plt.savefig('./complex_ratio.eps')
149156
plt.close()
150157

151158
plt.bar(bin_centers, all_zeros,
152159
width=(bins[1]-bins[0]), color='r', label='All zeros')
153160
plt.bar(bin_centers, real_zeros + near_real_zeros,
154161
width=(bins[1]-bins[0]), color='b', label='Real zeros')
155162
plt.legend(loc='best')
156-
ticks = np.arange(-max_plotval, max_plotval+1., 1)
163+
ticks = np.arange(-max_plotval, max_plotval+1., 2)
157164
tickvals = ["${}^{{{}}}$".format(int(np.sign(i+0.01)*10),int(abs(i)+cutoff_exp)) for i in ticks]
158165
ax = plt.gca()
166+
ax.ticklabel_format(style='sci')
159167
ax.set_xlim(left=bins[0], right=bins[-1])
160168
ax.set_xticks(ticks)
161-
ax.set_xticklabels(tickvals)
162-
plt.savefig('./complex_vertint_cutoff.png')
169+
ax.set_xticklabels(tickvals, fontsize=16)
170+
plt.yticks(fontsize=16)
171+
plt.savefig('./complex_vertint_cutoff.eps')
163172
plt.close()
164173

165174
plt.plot(bin_centers, (real_zeros + near_real_zeros) / all_zeros, color='k')
166-
ticks = np.arange(-max_plotval, max_plotval+1., 1)
175+
ticks = np.arange(-max_plotval, max_plotval+1., 2)
167176
tickvals = ["${}^{{{}}}$".format(int(np.sign(i+0.01)*10),int(abs(i)+cutoff_exp)) for i in ticks]
168177
ax = plt.gca()
178+
ax.ticklabel_format(style='sci')
169179
ax.set_xticks(ticks)
170-
ax.set_xticklabels(tickvals)
171-
plt.savefig('./complex_ratio_cutoff.png')
180+
ax.set_xticklabels(tickvals, fontsize=16)
181+
plt.yticks(fontsize=16)
182+
plt.savefig('./complex_ratio_cutoff.eps')
172183
plt.close()

count_clusters.py

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,24 @@
1+
#!/usr/bin/env python
2+
3+
import numpy as np
4+
import scipy.linalg as la
5+
import scipy.stats as stats
6+
import matplotlib
7+
matplotlib.use('Agg')
8+
import matplotlib.pyplot as plt
9+
import netCDF4 as nc4
10+
11+
CLUSTER_FILE_NAME = "/home/santos/Data/MG2_data_collection.10_cluster_labels.0001-01-06-00000.nc"
12+
13+
cfile = nc4.Dataset(CLUSTER_FILE_NAME, 'r')
14+
15+
ncluster = len(cfile.dimensions['ncluster'])
16+
17+
labels = cfile.variables["label"][:].flatten()
18+
19+
for i in range(ncluster):
20+
counter = 0
21+
for label in labels:
22+
if label == i:
23+
counter += 1
24+
print(counter, " grid points are in cluster ", i)

0 commit comments

Comments
 (0)