yt.geometry.grid_geometry_handler.GridIndex.partition_index_2d

GridIndex.partition_index_2d(axis)