Python Tree Map Thinking