Treemap Get Function In Python