yt.frontends.stream.data_structures.StreamParticleIndex.partition_index_2d

StreamParticleIndex.partition_index_2d(axis)