yt.frontends.athena.data_structures.AthenaHierarchy.partition_index_2d

AthenaHierarchy.partition_index_2d(axis)