python - 如何根据数组的两个日期添加列值？

Question

我是编程新手。我遇到了这个要求。我有一个数组，

data= ['2016-1-01', '2016-1-08', '2016-1-15', '2016-1-22', '2016-1-29', '2016-02-05', '2016-02-12', '2016-02-19', '2016-02-26']

我的查询结果如下:

date          a        b      c
2016-01-19    3        1      5
2016-01-20    10       4      5
2016-01-30    1        4      6

我正在尝试生成每周报告数据。

在此示例中，上述查询中的日期“2016-01-19”和“2016-01-20”位于数据数组的“2016-01-15”和“2016-01-22”之间，因此 a, b &c 待添加。

最终输出应该是这样的:

2016-1-01      0      0      0
2016-1-08      0      0      0
2016-1-15      13     5      10
2016-1-22      0      0      0
2016-1-29      1      4      6
2016-2-05      0      0      0
2016-2-12      0      0      0
2016-2-19      0      0      0
2016-2-26      0      0      0

Answer 1

假设 data 始终已排序并且没有重复元素(如果不是这种情况，您可以执行 data =sorted(set(data)) )，您可以做这样的事情:

import datetime

data = ['2016-1-01', '2016-1-08', '2016-1-15', '2016-1-22', '2016-1-29', '2016-02-05', '2016-02-12', '2016-02-19', '2016-02-26']
query = [(datetime.date(2016, 1, 19), 3, 1, 5), (datetime.date(2016, 1, 20), 10, 4, 5), (datetime.date(2016, 1, 30), 1, 4, 6)]

# Convert data to datetime objects
data = [datetime.datetime.strptime(d, '%Y-%m-%d').date() for d in data]

output = []
query_it = iter(query)
next_date = data[0]
next_nums = (0, 0, 0)
# Iterate through date ranges
for d_start, d_end in zip(data, data[1:] + [datetime.date.max]):
    # If the next interesting date is in range
    if next_date < d_end:
        nums = next_nums
        next_nums = (0, 0, 0)
        for q in query_it:
            q_date, q_nums = q[0], q[1:]
            if q_date < d_start:
                # Ignore dates before the first date in data
                continue
            elif q_date < d_end:
                # Add query numbers to count if in range
                nums = tuple(n1 + n2 for n1, n2 in zip(nums, q_nums))
            else:
                # When out of range save numbers for next
                next_date = q_date
                next_nums = q_nums
                break
    else:
        # Default to zero when no query dates in range
        nums = (0, 0, 0)
    # Add result to output
    output.append((d_start,) + nums)

for out in output:
    print(out)

输出:

(datetime.date(2016, 1, 1), 0, 0, 0)
(datetime.date(2016, 1, 8), 0, 0, 0)
(datetime.date(2016, 1, 15), 13, 5, 10)
(datetime.date(2016, 1, 22), 0, 0, 0)
(datetime.date(2016, 1, 29), 1, 4, 6)
(datetime.date(2016, 2, 5), 0, 0, 0)
(datetime.date(2016, 2, 12), 0, 0, 0)
(datetime.date(2016, 2, 19), 0, 0, 0)
(datetime.date(2016, 2, 26), 0, 0, 0)