这是一个快速的问题,我起初认为这很容易.一个小时,我不太确定!
所以,我有一个Python日期时间对象列表,我想绘制它们. x值是年和月,y值是此列表中本月发生的日期对象的数量.
也许一个例子会更好地证明这一点(dd / mm / yyyy):
[28/02/2018, 01/03/2018, 16/03/2018, 17/05/2018]
-> ([02/2018, 03/2018, 04/2018, 05/2018], [1, 2, 0, 1])
我的第一次尝试尝试按日期和年份分组,方法如下:
import itertools
group = itertools.groupby(dates, lambda date: date.strftime("%b/%Y"))
graph = zip(*[(k, len(list(v)) for k, v in group]) # format the data for graphing
正如您可能已经注意到的那样,这只会按列表中已存在的日期进行分组.在我上面的例子中,4月份没有发生日期的事实会被忽视.
接下来,我尝试查找开始日期和结束日期,并在它们之间循环:
import datetime
data = [[], [],]
for year in range(min_date.year, max_date.year):
for month in range(min_date.month, max_date.month):
k = datetime.datetime(year=year, month=month, day=1).strftime("%b/%Y")
v = sum([1 for date in dates if date.strftime("%b/%Y") == k])
data[0].append(k)
data[1].append(v)
当然,这只有在min_date.month小于max_date.month时才有效,如果它们跨越多年,则不一定如此.而且,它非常难看.
这样做有一种优雅的方式吗?
提前致谢
编辑:要清楚,日期是日期时间对象,而不是字符串.为了便于阅读,它们在这里看起来像字符串.
解决方法:
我建议使用pandas
:
import pandas as pd
dates = ['28/02/2018', '01/03/2018', '16/03/2018', '17/05/2018']
s = pd.to_datetime(pd.Series(dates), format='%d/%m/%Y')
s.index = s.dt.to_period('m')
s = s.groupby(level=0).size()
s = s.reindex(pd.period_range(s.index.min(), s.index.max(), freq='m'), fill_value=0)
print (s)
2018-02 1
2018-03 2
2018-04 0
2018-05 1
Freq: M, dtype: int64
s.plot.bar()
说明:
>首先从日期列表中创建Series
并转换为to_datetime
.
>按Series.dt.to_period
创建PeriodIndex
> groupby
by index(level = 0)并在GroupBy.size
之前获得计数
>按PeriodIndex
PeriodIndex
添加缺失期间,由索引的最大值和最小值创建
>最后的情节,例如对于酒吧 – Series.plot.bar