Python按日期过滤数据框中的行

Question

我有以下格式的数据框：

month     count
2015/01   100
2015/02   200
2015/03   300
...

并希望获得一个包含月份超过一个月的新数据框，例如2015/03。

我尝试使用以下代码：

sdf = df.loc[datetime.strptime(df['month'], '%Y-%m')>=datetime.date(2015,9,1,0,0,0)]

我是Python的新手。如果有人可以提供帮助，真的很感激。

Answer 1

例如，如果要获取月份大于2015/02的行，请使用pd.to_datetime而不是datetime.strptime，因为前者是矢量化的，并且可以接受Series对象作为参数（假设你正在使用熊猫）：

import pandas as pd
df[pd.to_datetime(df.month) >= pd.to_datetime("2015/02")]

#     month count
#1  2015/02   200
#2  2015/03   300