用matplotlib绘制更新的树状图

Question

我使用scipy.cluster.hierarchy中的dendogram函数生成树状图。我创建树形图并使用它：

linkage_matrix = ward(dist) 
dendrogram(linkage_matrix, orientation="left", labels=names) 

plt.tight_layout() 
plt.show() 

一切工作正常，直到现在。我想要做的是更新树状图的叶标记并绘制一个新的树形图。我尝试了两种方式：

1. 使用这里提到的leaf_label_func为：https://docs.scipy.org/doc/scipy-0.14.0/reference/generated/scipy.cluster.hierarchy.dendrogram.html

但这都不尽如人意。我得到一个错误，n不存在，这实际上是真的。函数中没有n。

1. 手动更新名称的树状图列表，然后将其添加到树状图。

`

linkage_matrix = ward(dist) 

dend = dendrogram(linkage_matrix, orientation="left", labels=names) 

leaf_names = dend['ivl'] 

for idx, item in enumerate(leaf_names): 
    if 'c' in item: 
     item = item+'_c' 
     leaf_names[idx] = item 

    if 'i' in item: 
     item = item+'_i' 
     leaf_names[idx] = item 

dend['ivl'] = leaf_names 


plt.tight_layout() 
plt.show() 

` 

然而，这正显示出剧情依然是树状图的旧版本，但如果我print(dend)，新叶的名字似乎是在那里。我在这里做错了什么？

Answer 1

可以适应你的需要更新dendogram的例子：

from matplotlib import pyplot as plt 
from scipy.cluster.hierarchy import dendrogram, linkage 
import numpy as np 
from pylab import * 
import time 
ion() 
for i in range(10): 
    #update here your dendogram values : 
    dist=[12.12,15.485,18.45,4.4,13.684,float(i+1)] 
    Z=linkage(dist) 
    # calculate it again : 
    dendrogram(Z, orientation="left") 
    # draw it again : 
    draw() 
    #wait 1 second between two update : 
    pause(1)