Как вывести время на ось графика matplotlib?

Question

[NV27]

Всем привет!

Имеется следующая задача: задан список со строковым значением времени (time_interval). Задан массив целевой функции (у). Хотелось бы построить график, где по оси x отложены значения из time_interval, а по оси y - значения целевой функции. Подскажите пожалуйста каким образом построить такой график.

import numpy as np
import matplotlib.pyplot as plt
time_interval = ['19:0:0', '19:1:0', '19:2:0', '19:3:0', '19:4:0']
y = np.random.randn(5)
x = np.array([x for x in range(5)])
plt.plot(time_interval, y)
plt.show()

Спасибо за внимание :)

Answer 1

[Александр]

Вам нужно перевести ваше время в тип datetime. И затем уже строить график.

Например:

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

fmt = dates.DateFormatter('%H:%M:%S')

fig, ax = plt.subplots()

time_interval = ['19:0:0', '19:1:0', '19:2:0', '19:3:0', '19:4:0']
time_interval = [dt.datetime.strptime(i, "%H:%M:%S") for i in time_interval]
y = np.random.randn(5)
x = np.array([x for x in range(5)])
ax.plot(time_interval, y, "-o")
ax.xaxis.set_major_formatter(fmt)
fig.autofmt_xdate()
plt.show()

Comments

[NV27]

Александр, да, примерно то, что мне нужно, спасибо!

Но возникла проблема, если временной интервал содержит конец одного дня и начало другого, то при построении графика, времена сортируются по возрастанию, и получается некорректная картинка.

Как исправить это? Пример прилагаю.

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

fmt = dates.DateFormatter('%H:%M:%S')

fig, ax = plt.subplots()

time_interval = ['23:30:00', '23:45:0', '0:0:0', '0:15:0', '0:30:00', '0:45:0', '01:00:00', '1:15:0', '1:30:00', '1:45:0', '2:00:00']
time_interval = [dt.datetime.strptime(i, "%H:%M:%S") for i in time_interval]
y = [71.94444444444444, 78.52631578947368, 74.3157894736842, 78.46428571428571, 73.4, 72.54166666666667, 73.17391304347827, 64.66666666666667, 80.05, 82.8, 79.2]
ax.plot(time_interval, y, "-o")
ax.xaxis.set_major_formatter(fmt)
fig.autofmt_xdate()
plt.show()

[Александр]

NV27: Решением будет добавить день. Конечно не очень элегантное решение.

time_interval = ['1 23:30:00', '1 23:45:0', '2 0:0:0', '2 0:15:0', '2 0:30:00', '2 0:45:0', '2 01:00:00', '2 1:15:0', '2 1:30:00', '2 1:45:0', '2 2:00:00']
time_interval = [dt.datetime.strptime(i, "%d %H:%M:%S") for i in time_interval]

[NV27]

Александр: Александр, да, все помогло. Спасибо большое. И, надеюсь, последний вопрос в рамках данного обсуждения: появилась необходимость построить график автокорреляционной функции по тем же данным, таким образом:

sm.graphics.tsa.plot_acf(pd.Series(y).values.squeeze())
Как в этом случае проставить время по горизонтальной оси?

[Александр]

Функция plot_acf не поддерживает пердачу значений по x, поэтому самый простой способ подписать время по оси x, будет такой:

import numpy as np
import matplotlib.pyplot as plt
from statsmodels.graphics.tsaplots import plot_acf

time_interval = ['19:00:00', '19:01:00', '19:02:00', '19:03:00', '19:04:00']
y = np.random.randn(5)

plt.figure()

ax = plt.gca()

plot_acf(y, ax)

plt.xticks(list(range(len(y))), time_interval)
plt.gcf().autofmt_xdate()

plt.tight_layout()
plt.show()