yt for Cosmology
- Accessing Cosmological Parameters
- Cosmological Coordinates
- Halo Finding
- Halo Profiling
In [1]:
# This will make sure all plots get showed in the notebook.
%matplotlib inline
import yt
In [13]:
ds = yt.load("../sample_data/Enzo_64/DD0043/data0043")
print ds
print ds.current_redshift
In [3]:
for k,v in ds.parameters.iteritems():
print "%s: %s" % (k, v)
In [14]:
ds2 = yt.load("../sample_data/Enzo_64/DD0033/data0033")
print ds2.current_redshift
In [15]:
## At z=0, the comoving Mpc/h should be equal to the proper Mpc/h:
print "ds:"
print ds.domain_width.in_units("Mpccm/h")
print ds.domain_width.in_units("Mpc/h")
## Here, at z-0.27, the comoving and proper box sizes should be different.
print "ds2:"
print ds2.domain_width.in_units("Mpccm/h")
print ds2.domain_width.in_units("Mpc/h")
In [16]:
p = yt.ProjectionPlot(ds, 'x', 'density', weight_field='density')
p.show()
In [17]:
import yt
from yt.analysis_modules.halo_analysis.api import HaloCatalog
hc = HaloCatalog(data_ds=ds, finder_method='hop')
hc.create()
In [18]:
ad = hc.halos_ds.all_data()
masses = ad['particle_mass'][:]
print 'Found %i halos, ranging from %e to %e' % (masses.size, masses.min(), masses.max())
print 'Found %i halos, ranging from %e to %e' % (masses.size, masses.min().in_units("Msun/h"), masses.max().in_units("Msun/h"))
In [19]:
p.annotate_halos(hc, annotate_field='particle_mass')
Out[19]:
In [21]:
hc = HaloCatalog(data_ds=ds, finder_method='hop')
# Filter out less massive halos
hc.add_filter("quantity_value", "particle_mass", ">", 1e14, "Msun")
# attach a sphere object to each halo whose radius extends
# to twice the radius of the halo
hc.add_callback("sphere", factor=2.0)
# use the sphere to calculate radial profiles of gas density
# weighted by cell volume in terms of the virial radius
hc.add_callback("profile", ["radius"],
[("gas", "overdensity")],
weight_field="cell_volume",
accumulation=False,
storage="virial_quantities_profiles")
hc.add_callback("virial_quantities", ["radius"],
profile_storage="virial_quantities_profiles")
hc.add_callback('delete_attribute', 'virial_quantities_profiles')
field_params = dict(virial_radius=('quantity', 'radius_200'))
hc.add_callback('sphere', radius_field='radius_200', factor=5,
field_parameters=field_params)
hc.add_callback('profile', ['virial_radius_fraction'],
[('gas', 'temperature')],
storage='virial_profiles',
weight_field='cell_mass',
accumulation=False, output_dir='profiles')
# Save the profiles
hc.add_callback("save_profiles", storage="virial_profiles",
output_dir="profiles")
hc.create()
In []:
!h5ls halo_catalogs/catalog/catalog.0.h5
In []:
hc
In []:
hc.halos_ds.field_list
In []: