Matplotlib - 根据它们在x轴上的位置设置不同的颜色

Question

我在matplotlib中创建了一个条形图，并且想要将条形的颜色更改为绿色或黄色，具体取决于它们在x上的位置-轴。例如，表示[5,6,8,9]中x值的数据的条应以黄色着色，而所有其他条应为绿色。我怎么做？谢谢！

Answer 1

首先，如果您发布了一些代码，那就太好了。我建议您阅读matplotlib.org上提供的信息，以了解您正在使用的函数的可能选项。下面是一个代码示例：

import numpy as np 
import matplotlib.pyplot as plt 
x = np.arange(10) 
xx = x**2 
cr = ['g','g','g','g','y','y','y','y','g','g'] 
fig,ax = plt.subplots() 
ax.bar(x,xx,color=cr) 
plt.show() 

这产生：

Answer 2

与上述方案中的唯一问题是我的是具有对依赖于图中的x轴杆不一致的次数。我通过基于我的x值生成颜色数组来绕过它。如果该值落在我想要着色的范围内 - 我在列表中添加了'r'，否则我添加'g'。但总的来说，萨沙的回答正是如此。

fig2, axs2 = plt.subplots(5, 2) 
fig2.suptitle("Plot A") 
plt.subplots_adjust(wspace = 0.5, hspace = 0.7) 
for j in range (10): 
    row = j/2 
    col = j % 2 
    box_title = box_sorted[j] 
    box_x = box_top10[j].keys() 
    box_y = box_top10[j].values() 
    colors = [] 
    for x in box_x: 
     if x in [5, 6, 8, 9]: 
      colors.append('r') 
     else: 
      colors.append('g') 
    axs2[row][col].bar([float(v) for v in box_x], [float(v) for v in box_y], align="center", color = colors) 
    axs2[row][col].set_title(box_title, fontsize = 12) 
    axs2[row][col].set_xlim ([0, 10.0]) 
    axs2[row][col].set_ylim ([0, float(max(box_y) + 3)]) 
    xticks = np.arange(0, 10, 1) 
    axs2[row][col].set_xticks(xticks) 
    yticks = np.arange(0, (max(box_y) + 5), int((max(box_y) + 6)/6)) 
    axs2[row][col].set_yticks(yticks) 
    axs2[row][col].tick_params(axis='both', which='major', labelsize=8) 
    axs2[row][col].set_xlabel("rating", fontsize = 8) 
    axs2[row][col].set_ylabel("#p", fontsize = 8) 

plt.show()