python添加缺少日期并更新相应的列表

时间:2018-06-05 20:22:18

标签: python python-2.7 defaultdict

我的dates列表中缺少日期,例如

['2018-06-01', '2018-06-02', '2018-06-03', '2018-06-06']

相对应的values列表
[3,5,3,7]

如何在排序列表中添加缺失日期,并为0

中的相应索引添加values

以上数值我从下面的数据中解析

data = defaultdict(Counter)
defaultdict(<class 'collections.Counter'>, {'2018-06-01': Counter({u'Values': 1}), '2018-06-03': Counter({u'Values': 2})}

如果我可以在defaultdict中添加缺少日期,那么也可以。

不重复我不想创建日期,我必须更新相应的值列表。

1 个答案:

答案 0 :(得分:2)

您可以根据数据创建真实的日期时间,开始和结束日期。然后,您可以创建所有可能的日期,并指定值0,然后更新现有的日期。

import datetime

dates = ['2018-06-01', '2018-06-02', '2018-06-03', '2018-06-06']
occ = [3,5,3,7]

startDate = datetime.datetime.strptime( dates[0], "%Y-%m-%d") # parse first date
endDate   = datetime.datetime.strptime( dates[-1],"%Y-%m-%d") # parse last date 
days = (endDate - startDate).days  # how many days between?

# create a dictionary of all dates with 0 occurences
allDates = {datetime.datetime.strftime(startDate+datetime.timedelta(days=k), 
                                       "%Y-%m-%d"):0 for k in range(days+1)}

# update dictionary with existing occurences (zip creates (date,number) tuples)
allDates.update(  zip(dates,occ) )

# sort the unsorted dict, decompose its items & zip then, wich generates your lists again
datesAfter,occAfter = map(list,zip(*sorted(allDates.items())))
print(datesAfter)
print(occAfter)

print(allDates)

输出:

['2018-06-01', '2018-06-02', '2018-06-03', '2018-06-04', '2018-06-05', '2018-06-06']
[3, 5, 3, 0, 0, 7]

{'2018-06-06': 7, 
 '2018-06-05': 0, 
 '2018-06-04': 0, 
 '2018-06-03': 3, 
 '2018-06-02': 5, 
 '2018-06-01': 3}

链接:zip()