根据范围内日期列的月份从 Pandas DataFrame 检索行

Question

我目前有一个名为Sales的表。 Sales表有一个名为sale_date的列，其格式YYYY-MM-DD ，我想提取月份在一个范围内的行。

| seller_id | product_id | buyer_id | sale_date    | quantity | price |
|-----------|------------|----------|--------------|----------|-------|
| 7         | 11         | 49       | '2019-01-21' | 5        | 3330  |
| 13        | 32         | 6        | '2019-02-10' | 9        | 1089  |
| 50        | 47         | 4        | '2019-01-06' | 1        | 1343  |

我试过类似的东西：

>>> df.loc[df['sale_date'].str.split('-').isin([1, 2, 3])]
>>> df.loc[[int(x[1]) for x in df['sale_date'].str.split('-')][1] in [1, 2, 3]]

但是这些分别导致类型错误和键错误。

有什么方法可以从sale_date列中提取月份并检查它是否在一个范围内？ 谢谢。

Answer 1

您可以将值转换为日期时间，然后提取月份：

df.loc[pd.to_datetime(df['sale_date']).dt.month.isin([1, 2, 3])]

或者通过索引str[1]并转换为整数来修改您的解决方案，从列表中提取第二个值：

df.loc[df['sale_date'].str.split('-').str[1].astype(int).isin([1, 2, 3])]