温馨提示×

如何在Matplotlib中绘制树状图或层次聚类图

小亿
129
2024-05-20 13:05:32
栏目: 编程语言
开发者测试专用服务器限时活动,0元免费领,库存有限,领完即止! 点击查看>>

在Matplotlib中绘制树状图或层次聚类图可以使用scipy库中的dendrogram函数来实现。以下是一个简单的示例代码:

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

# 生成随机数据
data = np.random.rand(10, 2)

# 计算层次聚类
Z = linkage(data, 'ward')

# 绘制树状图
plt.figure(figsize=(10, 5))
dendrogram(Z)
plt.title('Hierarchical Clustering Dendrogram')
plt.xlabel('Sample Index')
plt.ylabel('Distance')
plt.show()

在上面的代码中,首先生成了一些随机数据,然后使用linkage函数计算数据的层次聚类,并最后通过dendrogram函数绘制树状图。

您可以根据自己的数据和需求对代码进行修改和调整,以实现不同类型和样式的树状图或层次聚类图。

亿速云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>

推荐阅读:如何在Matplotlib中绘制堆积面积图

0