pynbody.filt.Annulus#
- class pynbody.filt.Annulus(r1: float | str | UnitBase, r2: float | str | UnitBase, cen: _SupportsArray[dtype[Any]] | _NestedSequence[_SupportsArray[dtype[Any]]] | bool | int | float | complex | str | bytes | _NestedSequence[bool | int | float | complex | str | bytes] = (0, 0, 0))[source]#
Bases:
And
A filter that selects particles in between two spheres specified by radii r1 and r2 centered on cen.
Methods
__call__
(sim)Return a boolean mask indicating which particles are in the filter.
cubic_cell_intersection
(centroids)Compute the intersection with cubic cells with the specified centroids.
where
(sim)Return the indices of particles that are in the filter.
- __init__(r1: float | str | UnitBase, r2: float | str | UnitBase, cen: _SupportsArray[dtype[Any]] | _NestedSequence[_SupportsArray[dtype[Any]]] | bool | int | float | complex | str | bytes | _NestedSequence[bool | int | float | complex | str | bytes] = (0, 0, 0))[source]#
Create an annulus filter.
- Parameters:
r1 – The inner radius of the annulus. If a string, it is interpreted as a unit string.
r2 – The outer radius of the annulus. If a string, it is interpreted as a unit string.
cen – The centre of the annulus. If a
pynbody.snapshot.simsnap.SimArray
, units can be provided and will be correctly accounted for.
- cubic_cell_intersection(centroids)#
Compute the intersection with cubic cells with the specified centroids.
This is currently used by the swift loader to figure out which cells to load
- where(sim: snapshot.SimSnap)#
Return the indices of particles that are in the filter.
This is a convenience method that is equivalent to np.where(f(sim)) but may be more efficient for some filters.