Python get_baryon_array Examples

Example #1

File: phase_plot.py Project: jgsuresh/DLA_script

def find_particle(part_id,num,base):
    """Gets the file and position of a particle with id part_id"""
    for i in range(0,500) :
        ids=hdfsim.get_baryon_array("ParticleIDs",num,base,i,dtype=np.int64)
        ind=np.where(ids==part_id)
        if np.size(ind) > 0:
            return (ind[0][0], i)

Example #2

File: phase_plot.py Project: sbird/DLA_script

def find_id(pos, num, base, snap_file=0):
    """Gets the id of a particle at some position"""
    ids = hdfsim.get_baryon_array("ParticleIDs",
                                  num,
                                  base,
                                  snap_file,
                                  dtype=np.int64)
    return ids[pos]

Example #3

File: phase_plot.py Project: sbird/DLA_script

def find_particle(part_id, num, base):
    """Gets the file and position of a particle with id part_id"""
    for i in range(0, 500):
        ids = hdfsim.get_baryon_array("ParticleIDs",
                                      num,
                                      base,
                                      i,
                                      dtype=np.int64)
        ind = np.where(ids == part_id)
        if np.size(ind) > 0:
            return (ind[0][0], i)

Example #4

File: phase_plot.py Project: jgsuresh/DLA_script

def find_id(pos,num,base,snap_file=0):
    """Gets the id of a particle at some position"""
    ids=hdfsim.get_baryon_array("ParticleIDs",num,base,snap_file,dtype=np.int64)
    return ids[pos]