在 Python (matplotlib) 中更改熱圖標簽的顏色

Question

我希望根據某些條件更改熱圖中ticklabels的顏色。 例如下面，我想更改包含A為red和B為green的ticklabels的顏色。

我嘗試制作一個列表['red', 'green', 'red', 'green']並將其傳遞給 color 選項，但到目前為止沒有成功。 任何幫助都會很棒。

import numpy as np
import matplotlib.pyplot as plt

alpha = ['ABC', 'BDF', 'ADF', 'BCF']

data = np.random.random((4,4))

fig = plt.figure()
ax = fig.add_subplot(111)
cax = ax.matshow(data, interpolation='nearest')
fig.colorbar(cax)

ax.set_xticklabels(['']+alpha)
ax.set_yticklabels(['']+alpha)

plt.show()

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應該是紅色還是綠色）？