Python 如何修改此代码以设置水平线的不同颜色？

如果是isUp，我想将水平颜色设置为黑色，如果是isDown，则设置为蓝色

levels = []
for i in range(2,df.shape[0]-2):
  if isUp(df,i):
    levels.append((i,df['low'][i]))
  elif isDown(df,i):
    levels.append((i,df['high'][i]))

for level in levels:
    plt.hlines(level[1],xmin=df['date'][level[0]],xmax=max(df['date']),colors='blue')
plt.show()

---

谢谢。

只有在检查isUp或isDown条件时，才能在第一个

循环中为
设置不同的颜色：

for i in range(2,df.shape[0]-2):
    if isUp(df,i):
        plt.hlines(df['low'][i],xmin=df['date'][i],xmax=max(df['date']),colors='black')
    elif isDown(df,i):
        plt.hlines(df['high'][i],xmin=df['date'][i],xmax=max(df['date']),colors='blue')

plt.show()