yt.frontends.athena.data_structures.AthenaGrid.partition_index_2d

AthenaGrid.partition_index_2d(axis)