結果 : making a heat map in python