yt.geometry.particle_geometry_handler.ParticleIndex.partition_index_3d

ParticleIndex.partition_index_3d(ds, padding=0.0, rank_ratio=1)