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

avatar
作者
猴君
阅读量:4

在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函数绘制树状图。

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

广告一刻

为您即时展示最新活动产品广告消息,让您随时掌握产品活动新动态!