python - 有没有办法将包含事件持续时间的日期索引数据帧转换为显示每天事件的二进制数据数据帧？

Question

对于措辞不当的问题，我深表歉意，但把它放在一行中是很受欢迎的。

我有一个日期索引数据框，其中包含与事件持续时间相关的数据，如下所示:

Date           Duration
12-01-2010     5
04-02-2010     1
14-02-2010     241
23-12-2010     6

我想把它变成一个数据框，每天索引，包含二进制数据，显示事件是否在给定的一天发生。例如，对于上面标识的第一个持续 5 天的事件:

Date           Event
12-01-2010     1
13-01-2010     1
14-01-2010     1
15-01-2010     1
16-01-2010     1
17-01-2010     0
18-01-2010     0

有什么想法吗？

谢谢

Answer 1

假设您使用的是 pandas 0.25，那么您可以使用 explode:

# Generate the list of days that has an event
s = df.apply(lambda row: pd.date_range(row['Date'], periods=row['Duration']), axis=1) \
        .explode() \
        .drop_duplicates()

# First line: we know those days have at least one event so mark them with 1
# Second line: expand it to cover every day of the year and fill the missing days with 0
result = pd.DataFrame({'Event': 1}, index=s) \
            .reindex(pd.date_range('2010-01-01', '2010-12-31'), fill_value=0)

结果:

            Event
2010-01-01      0
2010-01-02      0
2010-01-03      0
2010-01-04      0
2010-01-05      0
2010-01-06      0
2010-01-07      0
2010-01-08      0
2010-01-09      0
2010-01-10      0
2010-01-11      0
2010-01-12      1
2010-01-13      1
2010-01-14      1
2010-01-15      1
2010-01-16      1
2010-01-17      0
2010-01-18      0
2010-01-19      0
2010-01-20      0