Как изменить цвет части точек на графике?

Question

[milssky]

Есть код

ysmoothed = gaussian_filter1d(data, sigma=5)
x = [i for i in range(len(ysmoothed))]
y = [0.5 for i in range(len(ysmoothed))]
plt.plot(x, ysmoothed, 'b-', x, y, 'r-')
plt.show()

и получается картинка

Как сделать часть кривой, которая выше прямой y = 0.5 другим цветом?

Answer 1

[Developer]

if y>0.5:
      plt.plot(x, ysmoothed, 'b-', x, y, 'r-', color="#ff0000")
else:
      plt.plot(x, ysmoothed, 'b-', x, y, 'r-', color="#00ff00")

Comments

[milssky]

Не, не работает.

Traceback (most recent call last):
  File "C:\Program Files\JetBrains\PyCharm 2018.1.3\helpers\pydev\pydev_run_in_console.py", line 52, in run_file
    pydev_imports.execfile(file, globals, locals)  # execute the script
  File "C:\Program Files\JetBrains\PyCharm 2018.1.3\helpers\pydev\_pydev_imps\_pydev_execfile.py", line 18, in execfile
    exec(compile(contents+"\n", file, 'exec'), glob, loc)
  File "C:/Users/milssky/PycharmProjects/hrm/hrm.py", line 102, in <module>
    if y > 0.5:
TypeError: '>' not supported between instances of 'list' and 'float'

[elavusch]

milssky, идея норм: сначала делить списки по условию, а потом рисовать их разными цветами.

Answer 2

[lightmanLP]

Можно сохранить в файл и использовать PIL.