Pandas：在 dataframe 的列中過濾每個唯一單元格值的日期字段

Question

我有一個 dataframe df如下

YearMonth    CustID    Values
201901       12231     400
201902       12231     233
201903       12231     244
201904       12231     355
201901       12235     114
201902       12235     133
201903       12235     144
201904       12235     205

在上面，我有每個月（ YearMonth ）的CustID及其對應的值（ Values ）。

客觀的：

我想得到一個 dataframe ，其中每個客戶的數據將從201903即 2019 年 1 月到201901年即 2019 年 3 月。所以結果df看起來像

YearMonth    CustID    Values
201901       12231     400
201902       12231     233
201903       12231     244
201901       12235     114
201902       12235     133
201903       12235     144

我已經使用pd.to_datetime()將YearMonth轉換為datetime時間。

上面的dataframe怎么得到？ 我應該先過濾然后groupby(['CustID'])嗎？ 還是有其他出路？

Answer 1

簡單地說：

df = df[(df.YearMonth.ge(201901))&(df.YearMonth.le(201903))]

Answer 2

將pd.to_datetime與dt.month訪問器和series.between()一起使用：

df[pd.to_datetime(df.YearMonth,format='%Y%m').dt.month.between(1,3)]

---

   YearMonth  CustID  Values
0     201901   12231     400
1     201902   12231     233
2     201903   12231     244
4     201901   12235     114
5     201902   12235     133
6     201903   12235     144