只會柱狀圖、餅狀圖、折線圖怎麼行,來用Python畫個熱力圖

花花世界迷人眼,東西太多不知選?萬物存在即合理,不知根本,難得善法。術業有專攻,凡事不能蠻幹,不知怎麼畫熱力圖,請往下看。

只會柱狀圖、餅狀圖、折線圖怎麼行,來用Python畫個熱力圖


第一步:安裝並導入相關包

主要用到了pandas和pyecharts這兩個包,pandas用來讀取我存放在Excel中的數據,pyecharts用來將數據可視化。

這裡需要注意pyecharts的版本,因為不同版本之間的代碼可能有所不同,我安裝的是最新版,即1.7.1版本。

如果沒有這兩個包則先安裝:

<code>pip install pyecharts
pip install pandas/<code>

安裝完成後導入:

<code>import pandas as pd
from pyecharts import options as opts
from pyecharts.charts import HeatMap/<code>


第二步:讀取數據

我數據放在Excel中,如下:

只會柱狀圖、餅狀圖、折線圖怎麼行,來用Python畫個熱力圖

學生成績.xlsx

用pandas讀取數據:

<code>data = pd.read_excel('學生成績.xlsx')
x = data['姓名'].tolist()
y = data.columns.values[1:].tolist()
values = [[i, j, int(data.iloc[i, j+1])] for i in range(len(x)) for j in range(len(y))]/<code>

其中最後一行代碼作用是,將每一個成績與其對應的行列索引(也可以理解成座標)組合在一起。

我知道這麼說大家肯定不理解,所以我就直接將values值打印出來,如下:

<code>[[0, 0, 85], [0, 1, 74], [0, 2, 62], [0, 3, 72], [0, 4, 87], [0, 5, 80], [0, 6, 74], [0, 7, 94], [0, 8, 82], [1, 0, 72], [1, 1, 66], [1, 2, 71], [1, 3, 60], [1, 4, 66], [1, 5, 68], [1, 6, 68], [1, 7, 62], [1, 8, 93], [2, 0, 96], [2, 1, 80], [2, 2, 75], [2, 3, 85], [2, 4, 70], [2, 5, 83], [2, 6, 90], [2, 7, 75], [2, 8, 83], [3, 0, 78], [3, 1, 61], [3, 2, 70], [3, 3, 78], [3, 4, 89], [3, 5, 70], [3, 6, 65], [3, 7, 76], [3, 8, 95], [4, 0, 65], [4, 1, 81], [4, 2, 72], [4, 3, 96], [4, 4, 77], [4, 5, 88], [4, 6, 78], [4, 7, 90], [4, 8, 83], [5, 0, 70], [5, 1, 90], [5, 2, 65], [5, 3, 91], [5, 4, 85], [5, 5, 75], [5, 6, 70], [5, 7, 83], [5, 8, 72], [6, 0, 63], [6, 1, 72], [6, 2, 70], [6, 3, 80], [6, 4, 65], [6, 5, 80], [6, 6, 68], [6, 7, 93], [6, 8, 85]]/<code>


第三步:畫圖

<code>myHeatMap = HeatMap()
myHeatMap.add_xaxis(x)
myHeatMap.add_yaxis("學生成績", y, values)
myHeatMap.set_global_opts(title_opts=opts.TitleOpts(title="HeatMap"),
                          visualmap_opts=opts.VisualMapOpts(min_=60, max_=100))
myHeatMap.render(path='學生成績.html')/<code>

額,上面應該沒什麼好說的,大家看就懂。

其中60和100表示我數據的範圍,因為所有的成績都在60至100區間內。

將以上代碼運行後,就會得到一個html文件,打開後如下:

只會柱狀圖、餅狀圖、折線圖怎麼行,來用Python畫個熱力圖

學生成績.html


其實我事先準備了一些內容,用來敘述為什麼用熱力圖,而不用其它圖,但是寫出來後發現有點抽象,對讀者不太友好。

所以我只能讓各位自己私下細品了,各位可以想象一下用其它圖畫出來會是什麼樣,有沒有熱力圖直觀。

最後希望大家能品出什麼樣的數據適合什麼樣的圖...

關注微信公眾號“Python小鎮”,發現更多幹貨知識!


分享到:


相關文章: