如何创建表示每天多个时间间隔的图形

时间:2019-04-22 00:34:17

标签: python matplotlib graph

给出以下dict,其中包含每天的开放/关闭时间对:

timetable = {
    'monday': ['08:00', '12:00', '13:00', '18:00'],
    'tuesday': ['08:00', '12:00', '13:00', '18:00'],
    'wednesday': ['08:00', '12:00', '13:00', '18:00'],
    'thursday': ['08:00', '12:00', '13:00', '18:00'],
    'friday': ['08:00', '12:00', '13:00', '18:00', '19:00', '23:00'],
    'saturday': ['10:00', '16:00'],
    'sunday': ['10:00', '16:00'],
}

是否有一种方法来创建看起来像https://imgur.com/a/lK8CT9P的图形表示(这是通过gimp完成的,只是为了大致了解其外观)?

1 个答案:

答案 0 :(得分:0)

尝试这样的事情。

import matplotlib.pyplot as plt
import datetime
timetable = {
  'monday': ['08:00', '12:00', '13:00', '18:00'],
  'tuesday': ['08:00', '12:00', '13:00', '18:00'],
  'wednesday': ['08:00', '12:00', '13:00', '18:00'],
  'thursday': ['08:00', '12:00', '13:00', '18:00'],
  'friday': ['08:00', '12:00', '13:00', '18:00', '19:00', '23:00'],
  'saturday': ['10:00', '16:00'],
  'sunday': ['10:00', '16:00'],
}
COLOR_LIST = ['blue', 'red', 'green', 'pink', 'brown', 'orange', 'yellow']
HEIGHT = 2
fig, ax = plt.subplots()
y_parameter = 0
for day, day_data in timetable.iteritems():
    st = [datetime.datetime.strptime(i, "%H:%M") for i in day_data[::2]]
    et = [datetime.datetime.strptime(i, "%H:%M") for i in day_data[1::2]]
    color = COLOR_LIST.pop()
    for start_time, end_time in zip(st, et):
        diff = (end_time - start_time).total_seconds()
        x_paramter = datetime.timedelta(hours=start_time.hour, minutes=start_time.minute,
                                    seconds=start_time.second).total_seconds()
        ax.add_patch(plt.Rectangle((x_paramter, y_parameter), int(diff), HEIGHT, facecolor=color))
        centerx = x_paramter + 100
        centery = y_parameter + 1
        plt.text(centerx, centery, day, fontsize=5, wrap=True)
        plt.text(centerx, centery - 1, str(int(diff)), fontsize=5, wrap=True)
    ax.autoscale()
    ax.set_ylim(0, 24)
    y_parameter += 3
plt.show()

输出类似于:

enter image description here