Извлечение узлов и ребер из выходных данных иерархической кластеризации в Python - PullRequest
Новая
       87

Извлечение узлов и ребер из выходных данных иерархической кластеризации в Python

0 голосов
/ 05 марта 2019

Используя дневную историю цен акций, я рассчитал матрицу корреляции.

С помощью приведенного ниже кода я получаю простую древовидную дендрограмму.Но я хотел бы построить дендрограмму круга, упомянутую в этой ссылке. Иерархическое связывание ребер

Для любых других древовидных графов, кроме обычных, нам нужна информация Узлы и ребра .Как мы можем извлечь из кода scipy? 

from scipy.cluster.hierarchy import dendrogram, linkage
Z = linkage(corr, 'average')
Z[0]

#Plotting Dendrogram
plt.figure(figsize=(25, 10))
labelsize=20
ticksize=15
plt.title('Hierarchical Clustering Dendrogram for Standard Chartered', fontsize=labelsize)
plt.xlabel('stock', fontsize=labelsize)
plt.ylabel('distance', fontsize=labelsize)
dendrogram(
    Z,
    leaf_rotation=90.,  # rotates the x axis labels
    leaf_font_size=8.,  # font size for the x axis labels
    labels = corr.columns,
)
pylab.yticks(fontsize=ticksize)
pylab.xticks(rotation=-90, fontsize=ticksize)
plt.savefig('img/dendogram_'+index+'.png')
pl

t.show ()

...