yt.frontends.flash.data_structures.FLASHGrid.partition_index_2d

FLASHGrid.partition_index_2d(axis)