排定时间的matplotlib barh与实际工作时间

时间:2018-09-19 01:19:53

标签: python python-3.x matplotlib gantt-chart

我正在尝试为员工建立水平条形图,以显示给定日期的计划工作时间与实际工作时间。

我尝试了以下代码,但是正如您在下面的“图”图中所看到的,它是将实际工作时间(蓝色)连接到计划工作时间(绿色)的末尾。另外,x轴上的时间不是很清楚。

我要做的是为每个员工设置两个条形图,绿色条形图显示顶部的计划工作时间,蓝色条形图显示实际工作小时数,几乎就像一个甘特图。有人可以帮助我了解我的代码在哪里出问题吗?

#import stack
import pandas as pd
import datetime as dt
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.dates as mdates

#dummy df
df = pd.DataFrame([['Bob', '2018-09-14 9:00:00', '2018-09-14 18:00:00', 'actual']
                   , ['Bob', '2018-09-14 9:15:00', '2018-09-14 18:30:00', 'scheduled']
                   , ['Kim', '2018-09-14 9:00:00', '2018-09-14 18:00:00', 'actual']
                   , ['Kim', '2018-09-14 8:45:00', '2018-09-14 17:30:00', 'scheduled']]
                   , columns=['name','start','finish', 'type'])

#convert timestamp columns to datetime
df[['start', 'finish']] = df[['start', 'finish']].apply(pd.to_datetime)

#scheduled time period
scheduledStart = mdates.date2num(df['start'][(df['type'] == 'scheduled')].dt.to_pydatetime())
scheduledEnd =  mdates.date2num(df['finish'][(df['type'] == 'scheduled')].dt.to_pydatetime())
scheduledWidth = scheduledEnd - scheduledStart

#actual time period
actualStart = mdates.date2num(df['start'][(df['type'] == 'actual')].dt.to_pydatetime())
actualEnd =  mdates.date2num(df['finish'][(df['type'] == 'actual')].dt.to_pydatetime())
actualWidth = actualEnd - actualStart

#y axis values
yval = df['name'].unique()

#generate plot
fig, ax = plt.subplots()
ax.barh(yval, width = actualWidth, left = actualStart, color = 'blue', height = 0.3, label = 'actual')
ax.barh(yval, width = scheduledWidth, left = scheduledStart, color = 'green', height = 0.3, label = 'scheduled')

#format x axis to time of day
xfmt = mdates.DateFormatter('%H:%m')
ax.xaxis.set_major_formatter(xfmt)

# autorotate the dates
fig.autofmt_xdate()
plt.show()

figure

1 个答案:

答案 0 :(得分:0)

尝试使用不同的值,例如第一个[0, 1]和第二个[0.3, 1.3]的{​​{1}}(因此它们不会重叠)。我们只是在改变ax.barh的值。

height

然后更改ax.barh([0, 1], width = actualWidth, left = actualStart, color = 'blue', height = 0.3, label = 'actual') ax.barh([0.3, 1.3], width = scheduledWidth, left = scheduledStart, color = 'green', height = 0.3, label = 'scheduled') (在两个小节之间选择中心):

yticks

最终修复plt.yticks([0.15, 1.15], yval)

xtick

example_figure