yt.frontends.stream.data_structures.StreamParticleIndex.partition_index_3d

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