treemap color chart python - Search