Matplotlib: как отобразить линию с разными цветами на основе данных линии

Question

У меня есть массив Numpy, который принимает только два значения 0.0018 и 0.0018001

price_high_y = [0.0018 0.0018 0.0018 0.0018001 0.0018001 0.0018 0.0018 0.0018]

Я хотел бы отобразить эту строку со значениями 0,0018 черным и 0,0018001 желтым. Это должна быть горизонтальная линия. Я немного застрял .. Есть идеи? спасибо!

Answer 1

Это то, что вы хотите

price_high_y = np.array([0.0018, 0.0018, 0.0018, 0.0018001, 0.0018001, 0.0018, 0.0018, 0.0018])

yvals = sorted(np.unique(price_high_y))
colors = {0.0018: 'k', 0.0018001: 'y'}

for i, y in enumerate(yvals):
    plt.axhline(i+0.5, color=colors[y])

plt.yticks(np.arange(len(yvals))+0.5, yvals) 
plt.xticks([0, 1])
plt.show()