局部整体（六）利用python绘制树状图

import pandas as pd
from matplotlib import pyplot as plt
from scipy.cluster.hierarchy import dendrogram, linkage
import numpy as np
 
# 导入数据
url = 'https://raw.githubusercontent.com/holtzy/The-Python-Graph-Gallery/master/static/data/mtcars.csv'
df = pd.read_csv(url)
df = df.set_index('model')
df = df.reset_index(drop=True)

# 计算样本距离：对df执行层次聚类，并将结果存储在Z中
# Ward方差最小化算法用来计算样本距离
Z = linkage(df, 'ward')

# 树状图
dendrogram(Z, labels=df.index, leaf_rotation=90)

# 标题
plt.title('Hierarchical Clustering Dendrogram')
# 轴标签
plt.xlabel('sample index')
plt.ylabel('distance (Ward)')

plt.show()

# 基本树图

from pyecharts import options as opts
from pyecharts.charts import Tree

# 自定义数据
data = [
    {
        "children": [
            {"name": "B"},
            {
                "children": [{"children": [{"name": "I"}], "name": "E"}, {"name": "F"}],
                "name": "C",
            },
            {
                "children": [
                    {"children": [{"name": "J"}, {"name": "K"}], "name": "G"},
                    {"name": "H"},
                ],
                "name": "D",
            },
        ],
        "name": "A",
    }
]
c = (
    Tree()
    .add("", data)
    .set_global_opts(title_opts=opts.TitleOpts(title="基本树图"))
)
c.render_notebook()

import pandas as pd
from matplotlib import pyplot as plt
from scipy.cluster.hierarchy import dendrogram, linkage, set_link_color_palette
import numpy as np

plt.rcParams['font.sans-serif'] = ['SimHei'] # 用来正常显示中文标签
 
# 导入数据
url = 'https://raw.githubusercontent.com/holtzy/The-Python-Graph-Gallery/master/static/data/mtcars.csv'
df = pd.read_csv(url)
df = df.set_index('model')

# 计算样本距离
Z = linkage(df, 'ward')
 

# 初始化布局
fig = plt.figure(figsize=(12,12))

# 自定义标签
plt.subplot(3, 2, 1) 
dendrogram(Z, leaf_rotation=90, leaf_font_size=8, labels=df.index)
plt.title('自定义标签')

# 自定义颜色和颜色阈值
plt.subplot(3, 2, 2) 
set_link_color_palette(['#b30000','#996600', '#b30086']) # 颜色
dendrogram(Z, color_threshold=240, above_threshold_color='grey') # 颜色阈值
plt.axhline(y=240, c='grey', lw=1, linestyle='dashed')
plt.title('自定义颜色和颜色阈值')

# 簇截断-叶子数量
plt.subplot(3, 2, 3) 
dendrogram(Z, truncate_mode = 'lastp', p=4 ) 
plt.title('簇截断-叶子数量')

# 簇截断-显示层数
plt.subplot(3, 2, 4) 
dendrogram(Z, truncate_mode = 'level', p=2 ) 
plt.title('簇截断-显示层数')

# 水平方向-右
plt.subplot(3, 2, 5) 
dendrogram(Z, orientation="right", labels=df.index) 
plt.title('水平方向-右')
    
# 水平方向-左
plt.subplot(3, 2, 6) 
dendrogram(Z, orientation="left", labels=df.index) 
plt.title('水平方向-左')


fig.tight_layout() # 自动调整间距
plt.show()