Python groupby 日期但带有 if 语句

Question

我的数据如下所示:

id   Open    Close
1    1/1/15   1/1/15 
2    1/1/15   2/1/15
3    3/1/15   4/1/15  

我需要创建一个数据框来显示任何一天的未结案件数，因此上面数据的结果如下所示:

Date   #Open
1/1/15   1
2/1/15   0
3/1/15   1

有什么想法吗？

Answer 1

此方法创建第一个打开的案例与最后一个打开或关闭的案例的最大天数之间的所有天数的索引。然后，它会迭代每个日期并筛选相关日期的数据帧，检查结果大小。

df['Open'] = pd.to_datetime(df.Open)
df['Close'] = pd.to_datetime(df.Close)

idx = pd.date_range(df.Open.min(), max(df.Open.max(), df.Close.max()))

cases = pd.DataFrame([len(df[(date >= df.Open) & (date < df.Close)]) 
                      for date in idx], 
                     index=idx, columns=['case_count'])

>>> cases.head(3)
            case_count
2015-01-01           1
2015-01-02           1
2015-01-03           1

>>> cases.tail(3)
            case_count
2015-03-30           1
2015-03-31           1
2015-04-01           0