yt.frontends.stream.data_structures.StreamGrid.partition_index_2d

StreamGrid.partition_index_2d(axis)