python - 根据字典 matplotlib 中的值更改标签颜色

Question

我有一个字典，其中包含作为标签的键和作为绘图颜色的值。使用以下示例:Change color of specific ticks at plot with matplotlib 。如何根据字典中的键更改标签颜色。请注意，标签可以在 x 轴上重复多次，并且可以是字符串，例如 labels = 'A'、'B'、'C'、'D'。

这是代码的一点修改:

import matplotlib.pyplot as plt

fig, ax = plt.subplots(figsize=(5,4))
ax.plot([1,2,3])
colors = {0.00 :'grey',0.25:'yellow',0.50:'brown',2.00:'red'}
for k, v in colors.items():
    ax.get_xticklabels()[k].set_color(v)

plt.show()

我希望绘图上的每个标签都与一种颜色相关联，但我得到一个带有此错误的普通绘图:TypeError:列表索引必须是整数或切片，而不是 float 。我正在尝试将其专门适应字符串标签。

Answer 1

一种解决方案是检查字典键中是否存在刻度标签。为了比较确切的值，我将键从浮点转换为字符串。我相信这对你来说应该不是问题。

P.S.:根据@ImportanceOfBeingEarnest 的评论，我想说这个答案假设该数字保持静态且未调整大小。

import matplotlib.pyplot as plt

fig, ax = plt.subplots(figsize=(5,4))
ax.plot([1,2,3])
colors = {'0.00' :'grey','0.25':'yellow','0.50':'brown','2.00':'red'}

fig.canvas.draw()

for xtic in ax.get_xticklabels():
    if xtic.get_text() in colors.keys(): # Change color if exist else not
        xtic.set_color(colors[xtic.get_text()])

plt.show()