python - 计算使用 Pandas 的独特天数？

Question

我有以下 df:

            h1 h2         date  c
0   002599f58e  A   2019-09-23  1
1   002599f58e  A   2019-09-24  3
2   005344fe83  C   2019-09-30  7
3   005344fe83  D   2019-10-02  1
4   005344fe83  D   2019-10-03  2

我想计算 独特天数 h1 在 h2 中。所以，我应该以这样的方式结束:

            h1  h2  count
0   002599f58e   A      2
2   005344fe83   C      1
4   005344fe83   D      2

因为 A 在不同的两天出现，C 在一天出现，D 在另外两天出现。我试过类似的东西

days = df.groupby(["h1", 
                   df["date"].dt.floor("D"),
                   "h3"]).size().reset_index(name="count")

但这行不通。有什么想法吗？

Answer 1

使用DataFrameGroupBy.nunique :

#convert to datetimes and if necessary remove times by floor
df['date'] = pd.to_datetime(df['date']).dt.floor("D")

days = df.groupby(["h1", 'h2'])["date"].nunique().reset_index(name="count")
print (days)
           h1 h2  count
0  002599f58e  A      2
1  005344fe83  C      1
2  005344fe83  D      2