python提取出月份,然后将月份进行分段,将不同的月份划分到不同的季节中,最终得到一个新的 season 列。
时间: 2024-05-06 22:16:26 浏览: 53
可以使用Python的datetime模块来提取月份和判断季节,具体实现如下:
```python
import pandas as pd
from datetime import datetime
# 创建测试数据
df = pd.DataFrame({'date': ['2022-01-01', '2022-02-15', '2022-05-20', '2022-09-10', '2022-12-05']})
# 将字符串类型的日期转换成datetime类型
df['date'] = pd.to_datetime(df['date'])
# 提取月份
df['month'] = df['date'].dt.month
# 判断季节
def get_season(month):
if month in [3, 4, 5]:
return '春季'
elif month in [6, 7, 8]:
return '夏季'
elif month in [9, 10, 11]:
return '秋季'
else:
return '冬季'
df['season'] = df['month'].apply(get_season)
print(df)
```
输出结果如下:
```
date month season
0 2022-01-01 1 冬季
1 2022-02-15 2 冬季
2 2022-05-20 5 春季
3 2022-09-10 9 秋季
4 2022-12-05 12 冬季
```
其中,`get_season()`函数根据月份返回对应的季节,然后使用`apply()`方法将函数应用到每一行的月份上,得到最终的季节数据。
阅读全文