Я выполнил иерархическую кластеризацию, используя scipy, и следующая моя дендрограмма. Теперь мне нужно получить надписи на листьях определенного цвета. Например: мне нужно найти метки для красного, синего, зеленого и т. Д. Любая помощь по этому поводу?
B = dendrogram(linkage_matrix,
color_threshold=250,
labels=df.session.tolist(),
distance_sort='ascending')
Мне просто нужно было решить ту же проблему, но ответ выше неверен. Цвета из color_list соответствуют не цветам листьев, а цветам ссылок - пи-образных структур, соединяющих кластеры. Color_list и ivl даже не обязательно должны быть одинаковой длины.
Здесь вы можете найти способ получить метки листьев: http://nxn.se/post/90198924975/extract-cluster-elements-by-color-in-python
B или что-то еще, возвращаемое функцией дендрограммы, представляет собой словарь со следующими ключами:
print B.keys()
['ivl', 'dcoord', 'leaves', 'color_list', 'icoord']
color_list будет выглядеть примерно так:
print B['color_list']
['g', 'g', 'b', 'r', 'r', 'r', 'r', 'b']
И ivl будет выглядеть так:
print B['ivl']
['4', '1', '5', '6', '3', '2', '7', '0']
Осталось только составить список:
myInd = [i for i, c in zip(B['ivl'], B['color_list']) if c=='g']
print myInd
['4', '1']
