Как выделить области на графике, чтобы указать скольжение windows в Python? - PullRequest
Новая
       37

Как выделить области на графике, чтобы указать скольжение windows в Python?

2 голосов
/ 09 января 2020

У меня есть временной ряд с экстремальными событиями, и я попытался определить ширину этих экстремальных событий, используя метод скользящих окон. Я использовал код: 

def moving_window(s, length, step =1):
       streams = it.tee(s, length)
       return zip(*[it.islice(stream, i, None, step*length) for stream, i in zip(streams, it.count(step=step))])
x_=list(moving_window(s, 15))
x_=np.asarray(x_) #windows
print(x_)

, и у меня есть выход для временного ряда: Zoomed in time series containing one extreme event

[[  0   1   2   3   4   5   6   7   8   9  10  11  12  13  14]
 [ 15  16  17  18  19  20  21  22  23  24  25  26  27  28  29]
 [ 30  31  32  33  34  35  36  37  38  39  40  41  42  43  44]
 [ 45  46  47  48  49  50  51  52  53  54  55  56  57  58  59]
 [ 60  61  62  63  64  65  66  67  68  69  70  71  72  73  74]
 [ 75  76  77  78  79  80  81  82  83  84  85  86  87  88  89]
 [ 90  91  92  93  94  95  96  97  98  99 100 101 102 103 104]
 [105 106 107 108 109 110 111 112 113 114 115 116 117 118 119]
 [120 121 122 123 124 125 126 127 128 129 130 131 132 133 134]
 [135 136 137 138 139 140 141 142 143 144 145 146 147 148 149]
 [150 151 152 153 154 155 156 157 158 159 160 161 162 163 164]
 [165 166 167 168 169 170 171 172 173 174 175 176 177 178 179]
 [180 181 182 183 184 185 186 187 188 189 190 191 192 193 194]]

Я хочу выделить скользящее windows с картой цветов. Мне нужно что-то вроде следующего изображения: This contains 20 images but just consider one image.

Я хочу знать, как использовать цветовую карту для этого (в изображении 20 временных рядов, но рассмотрим только один .). Кто-нибудь может помочь?

1 Ответ

2 голосов
/ 09 января 2020

Вот пример использования функции синуса для демонстрации концепции. axvspan dr aws вертикальные пролеты. Цвет может быть установлен из цветовой карты. color=0 будет слева от карты, color=1 полностью справа. Здесь используются красные. Некоторые эксперименты с альфа и индексами предполагают, что alpha=0.6 и индексы 0,75 и снижение дают некоторые цвета, аналогичные приведенным в примере. 

import matplotlib.pyplot as plt
import numpy as np

x_min = 0
x_max = 120
x = np.linspace(x_min, x_max, 10000)
y = np.sin(x/3)

fig, ax = plt.subplots(figsize=(12,2))
ax.plot(x, y, color='royalblue')

cmap = plt.cm.Reds # e.g. plt.cm.plasma_r or plt.cm.YlOrRd also seem interesting
current_x = 105
x_step = 16
for i in range(8):
    ax.axvspan(current_x - (i + 1) * x_step, current_x - i * x_step,
               alpha=0.6, color=cmap(0.75 - i / 20))
ax.set_xlim(x_min, x_max)
plt.tight_layout()
plt.show()

example plot

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

for i in range(8):
    ax.axvspan(current_x - (i + 1) * x_step, current_x - i * x_step,
               alpha=0.5 - i / 20, color='red')

Конечно, можно варьировать как альфа, так и цвет вместе для более точной настройки. Нужны некоторые эксперименты, чтобы найти цвета, которые достаточно различны и не слишком кричат.

Вот пример с cmap = plt.cm.inferno_r и ax.axvspan(..., alpha=0.4, color=cmap(0.8 - i / 10)):

another example plot

...