在Python中更改热图标签的颜色（matplotlib）

Question

我希望根据某些条件更改热图中ticklabels的颜色。例如，我想将ticklabels包含A的{​​{1}}和red的颜色更改为B。

我尝试制作一个列表green并将其传递给颜色选项但到目前为止没有成功。任何帮助都会很棒。

['red', 'green', 'red', 'green']

Answer 1

您可以使用标签文本对象上的set_color设置文本颜色：

for l in ax.xaxis.get_ticklabels()+ax.yaxis.get_ticklabels():
    if 'A' in l.get_text():
        l.set_color('r')
    elif 'B' in l.get_text():
        l.set_color('g')

当然，如果标签中存在A和B（即ABC应该是红色还是绿色），这并不能解决会发生什么情况？