python - 如何修改此代码为水平线设置不同的颜色？

Question

如果是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()

谢谢。

标签： python

Answer 1

for只有在检查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()