yt.frontends.stream.data_structures.StreamHierarchy.partition_index_2d

StreamHierarchy.partition_index_2d(axis)