Plotly绘制树状热力图（treemap）Plotly实例教程

Posted 2022-02-03 K同学啊

从DataFrame获取数据绘制

import plotly.express as px
import pandas as pd

#任意的多组列表
a = [5,8,16,32,64,128,256,256,5,8,16,32,64,128,256,256]
b = ["a","g","j","i","g","f","h","b","a","c","c","a","g","f","g","g"]

#字典中的key值即为csv中的列名
data_dict = 'a_name':a,'text':b

#设置DataFrame中列的排列顺序
#如果无需重命名，columns这一参数可以省略
dataFrame = pd.DataFrame(data_dict, columns=['a_name', 'text'])

dataFrame.head(3)

b = dataFrame['text'].value_counts().index.tolist()
a = dataFrame['text'].value_counts().tolist()
print(a,b)

row = pd.DataFrame('scenario' : [])
row["scenario"] = b
row["Number"]   = a
fig = px.treemap(row, path= ["scenario"], values="Number",title='Tree of Mentions')
fig.show()

从列表List中获取数据绘制

import plotly.express as px
import pandas as pd

a = [5,8,16,32,64,128,256,256,5,8,16,32,64,128,256,256]
b = ["a","g","j","i","g","f","h","b","a","c","c","a","g","f","g","g"]

row = pd.DataFrame('scenario' : [])
row["scenario"] = b
row["Number"]   = a
fig = px.treemap(row, path= ["scenario"], values="Number",title='Tree of Mentions')
fig.show()