結果 : how to code a heat map in python