树状图默认的黄色和绿色有点丑,请问怎么修改颜色

img

做出的树状图默认颜色是黄色和绿色,请问怎么修改,还有就是纵轴直接从4开始可以吗?还是说0-4之间必须加上波浪线表示省略?

使用link_color_func参数,设定链接颜色。示例:

from scipy.cluster import hierarchy
import matplotlib.pyplot as plt
import numpy as np
ytdist = np.array([662., 877., 255., 412., 996., 295., 468., 268., 400., 754., 564., 138., 219., 869., 669.])
colors = ['r', 'b','g']*5
Z = hierarchy.linkage(ytdist, 'single')
plt.figure()
dn = hierarchy.dendrogram(Z,link_color_func=lambda k: colors[k])
plt.show()


看看这个吧
https://vimsky.com/examples/usage/python-scipy.cluster.hierarchy.dendrogram.html