yt.geometry.particle_geometry_handler.ParticleIndex.partition_index_2d

ParticleIndex.partition_index_2d(axis)