怎么用Python畫熱力圖

這篇文章主要介紹“怎么用Python畫熱力圖”，在日常操作中，相信很多人在怎么用Python畫熱力圖問題上存在疑惑，小編查閱了各式資料，整理出簡單好用的操作方法，希望對大家解答”怎么用Python畫熱力圖”的疑惑有所幫助！接下來，請跟著小編一起來學(xué)習(xí)吧！

第一步：安裝并導(dǎo)入相關(guān)包

主要用到了pandas和pyecharts這兩個(gè)包，pandas用來讀取我存放在Excel中的數(shù)據(jù)，pyecharts用來將數(shù)據(jù)可視化。

這里需要注意pyecharts的版本，因?yàn)椴煌姹局g的代碼可能有所不同，我安裝的是最新版，即1.7.1版本。

如果沒有這兩個(gè)包則先安裝：

pip install pyecharts pip install pandas

安裝完成后導(dǎo)入：

import pandas as pd from pyecharts import options as opts from pyecharts.charts import HeatMap

第二步：讀取數(shù)據(jù)

我數(shù)據(jù)放在Excel中，如下：

學(xué)生成績.xlsx

用pandas讀取數(shù)據(jù)：

data = pd.read_excel('學(xué)生成績.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))]

其中最后一行代碼作用是，將每一個(gè)成績與其對應(yīng)的行列索引(也可以理解成坐標(biāo))組合在一起。

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

[[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]]

第三步：畫圖

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

額，上面應(yīng)該沒什么好說的，大家看就懂。

其中60和100表示我數(shù)據(jù)的范圍，因?yàn)樗械某煽兌荚?0至100區(qū)間內(nèi)。

將以上代碼運(yùn)行后，就會(huì)得到一個(gè)html文件，打開后如下:

學(xué)生成績.html

到此，關(guān)于“怎么用Python畫熱力圖”的學(xué)習(xí)就結(jié)束了，希望能夠解決大家的疑惑。理論與實(shí)踐的搭配能更好的幫助大家學(xué)習(xí)，快去試試吧！若想繼續(xù)學(xué)習(xí)更多相關(guān)知識，請繼續(xù)關(guān)注億速云網(wǎng)站，小編會(huì)繼續(xù)努力為大家?guī)砀鄬?shí)用的文章！