фон диаграммы matplotlib в зависимости от данных

Question

Я пытаюсь построить некоторые данные и хочу иметь цветной фон в зависимости от данных.

В следующем примере я хочу, чтобы data1 и data2 были на левой оси и data3 на правой оси.Это работает.Но дополнительно я попытался раскрасить фон в зависимости от данных3.

Как мне нужно отформатировать данные, чтобы они работали?

import matplotlib.pyplot as plt
from datetime import datetime as dt
import matplotlib.dates as md

fig, ax1 = plt.subplots(constrained_layout=True)

data1 = [51.2, 51.2, 51.2, 50.7, 50.7, 50.5, 50.4, 50.7, 50.6]
data2 = [46.5, 46.1, 46.2, 46.3, 46.4, 46.3, 46.2, 46.1, 45.5]
data3 = [ 0.0,  1.0,  1.0,  1.0,  0.0,  0.0,  0.0,  0.0,  0.0]

timestamps = [1524614516, 1524615134, 1524615587, 1524615910, 1524616235, 1524616559, 1524616866, 1524617189, 1524617511]
timestamps_ = [dt.utcfromtimestamp(x) for x in timestamps]

for data in (data1,data2):
    ax1.plot(timestamps_, data, marker='.', linestyle='-')
ax1.set_ylabel("degC")

ax2 = ax1.twinx()
ax2.plot(timestamps_, data3, marker='x', linestyle='-')
ax2.pcolor(ax2.get_xlim(), ax2.get_ylim(), zip(timestamps_, data3), cmap='RdGn', alpha=0.3) 
ax2.set_ylabel("ON OFF")       

ax1.set_title("Mytitle")
for tick in ax1.xaxis.get_major_ticks():
    tick.label1.set_horizontalalignment('right')
    tick.label1.set_rotation(35)
xfmt = md.DateFormatter('%Y-%m-%d %H:%M:%S')
ax1.xaxis.set_major_formatter(xfmt)

plt.show()

Сообщение об ошибке:

Traceback (most recent call last):
  File "/home/tobias/workspace/python_pyplot_test/main.py", line 25, in <module>
    ax2.pcolor(ax2.get_xlim(), ax2.get_ylim(), zip(timestamps_, data3), cmap='RdGn', alpha=0.3) 
  File "/usr/local/lib/python2.7/dist-packages/matplotlib/__init__.py", line 1855, in inner
    return func(ax, *args, **kwargs)
  File "/usr/local/lib/python2.7/dist-packages/matplotlib/axes/_axes.py", line 5732, in pcolor
    X, Y, C = self._pcolorargs('pcolor', *args, allmatch=False)
  File "/usr/local/lib/python2.7/dist-packages/matplotlib/axes/_axes.py", line 5576, in _pcolorargs
C.shape, Nx, Ny, funcname))
TypeError: Dimensions of C (9, 2) are incompatible with X (2) and/or Y (2); see help(pcolor)

Answer 1

Другой вариант - использовать axvspan с :

Одна разница между использованием axvspan и pcolor означает, что вертикальный промежуток (прямоугольники), нарисованный axvspan, не ограничен в y -направлении, а прямоугольники pcolor - нет.Поэтому, если вы используете кнопку zoom для изменения размера графика, прямоугольники axvspan будут растягиваться до бесконечности (грубо говоря), а при уменьшении прямоугольников pcolor будут видны белые области.Это не имеет большого значения, просто подумал, что вы хотели бы знать.

---

Также обратите внимание, что если вертикальные пролеты начинаются с первой точки данных и продолжаются до следующей точки данных, то последнее значение вdata3 никогда не привыкает.(Девять точек данных составляют восемь вертикальных интервалов).Однако если вы центрируете вертикальные промежутки вокруг точек данных - поэтому каждая точка данных находится в центре промежутка, тогда можно использовать все 9 значений в data3.

Раскомментируйте прокомментированный код ниже(и закомментируйте текущее определение timestamps_left и timestamps_right), чтобы увидеть разницу.

---

import numpy as np
import matplotlib.pyplot as plt
from datetime import datetime as dt
import matplotlib.dates as md

def topydates(timestamps):
    return [dt.utcfromtimestamp(x) for x in timestamps]
fig, ax1 = plt.subplots(constrained_layout=True)

data1 = [51.2, 51.2, 51.2, 50.7, 50.7, 50.5, 50.4, 50.7, 50.6]
data2 = [46.5, 46.1, 46.2, 46.3, 46.4, 46.3, 46.2, 46.1, 45.5]
data3 = [ 0.0,  1.0,  1.0,  1.0,  0.0,  0.0,  0.0,  0.0,  1.0]

timestamps = np.array([1524614516, 1524615134, 1524615587, 1524615910,
                       1524616235, 1524616559, 1524616866, 1524617189, 1524617511])
timestamps_ = topydates(timestamps)

for data in (data1,data2):
    ax1.plot(timestamps_, data, marker='.', linestyle='-')
ax1.set_ylabel("degC")

ax2 = ax1.twinx()
ax2.plot(timestamps_, data3, marker='x', linestyle='-')

# if you want the axvspans to be centered around the data points
# widths = np.diff(timestamps)
# midpoints = timestamps[:-1] + widths/2.0
# timestamps_left = topydates(np.r_[timestamps[0]-widths[0]/2, midpoints])
# timestamps_right = topydates(np.r_[midpoints, timestamps[-1] + widths[-1]/2.0])
# if you uncomment the code above, then comment-out the line below:
timestamps_left, timestamps_right = timestamps_[:-1], timestamps_[1:]

cmap = plt.get_cmap('RdYlGn')
for left, right, val in zip(timestamps_left, timestamps_right, data3):
    print(left, right)
    color = cmap(val)
    ax2.axvspan(left, right, facecolor=color, alpha=0.3)

ax2.set_ylabel("ON OFF")       

ax1.set_title("Mytitle")
for tick in ax1.xaxis.get_major_ticks():
    tick.label1.set_horizontalalignment('right')
    tick.label1.set_rotation(35)
xfmt = md.DateFormatter('%Y-%m-%d %H:%M:%S')
ax1.xaxis.set_major_formatter(xfmt)

plt.show()

Answer 2

Вот минимальное решение для того, что вы хотите:

import matplotlib.pyplot as plt
from datetime import datetime as dt
import matplotlib.dates as md
import numpy as np


data3 = np.array([ 0.0,  1.0,  1.0,  1.0,  0.0,  0.0,  0.0,  0.0,  0.0])

x=np.arange(9)
xp,yp=np.meshgrid(x,data3)
xp=xp.astype(float)-0.5
bgcolor=np.ones(xp.shape)*data3[None,:]

plt.pcolor(xp,yp,bgcolor) 
plt.plot(x, data3, marker='x', linestyle='-')

Я вынул вторую ось и все галочки, поскольку они не были связаны с самой проблемой.