Python：截止日期

Question

我想将日期时间序列转换为季节，例如对于第3、4、5个月，我想将其替换为2（春季）；对于6、7、8个月，我想用3（夏季）等替换它们。

所以，我有这个系列

id
1       2011-08-20
2       2011-08-23
3       2011-08-27
4       2011-09-01
5       2011-09-05
6       2011-09-06
7       2011-09-08
8       2011-09-09
Name: timestamp, dtype: datetime64[ns]

这是我一直在尝试使用的代码，但无济于事。

# Get seasons
spring = range(3, 5)
summer = range(6, 8)
fall = range(9, 11)
# winter = everything else

month = temp2.dt.month
season=[]

for _ in range(len(month)):
    if any(x == spring for x in month):
       season.append(2) # spring 
    elif any(x == summer for x in month):
        season.append(3) # summer
    elif any(x == fall for x in month):
        season.append(4) # fall
    else:
        season.append(1) # winter

和

for _ in range(len(month)):
    if month[_] == 3 or month[_] == 4 or month[_] == 5:
        season.append(2) # spring 
    elif month[_] == 6 or month[_] == 7 or month[_] == 8:
        season.append(3) # summer
    elif month[_] == 9 or month[_] == 10 or month[_] == 11:
        season.append(4) # fall
    else:
        season.append(1) # winter

这两种解决方案均无效，特别是在第一个实现中，我收到一个错误：

ValueError：具有多个元素的数组的真值是暧昧。使用a.any（）或a.all（）

尽管第二个列表中有很多错误。有什么想法吗？谢谢

Answer 1

您可以使用简单的数学公式将一个月压缩为一个季节，例如：

>>> [(month%12 + 3)//3 for month in range(1, 13)]
[1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 4, 1]

因此对于使用矢量操作的用例（@DSM信用）：

>>> (temp2.dt.month%12 + 3)//3
1    3
2    3
3    3
4    4
5    4
6    4
7    4
8    4
Name: id, dtype: int64

Answer 2

也可以使用字典映射。

1. 创建一个将一个月映射到一个季节的字典：

   In [27]: seasons = [1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 4, 1]
   
   In [28]: month_to_season = dict(zip(range(1,13), seasons))
   
   In [29]: month_to_season 
   Out[29]: {1: 1, 2: 1, 3: 2, 4: 2, 5: 2, 6: 3, 7: 3, 8: 3, 9: 4, 10: 4, 11: 4, 12: 1}
   

2. 使用它可以将月份转换为季节

   In [30]: df.id.dt.month.map(month_to_season) 
   Out[30]: 
   1    3
   2    3
   3    3
   4    4
   5    4
   6    4
   7    4
   8    4
   Name: id, dtype: int64
   

---

性能：相当快

In [35]: %timeit df.id.dt.month.map(month_to_season) 
1000 loops, best of 3: 422 µs per loop

Answer 3

我认为这会起作用。

while True:
date=int(input("Date?"))
season=""
if date<4:
    season=1
elif date<7:
    season=2
elif date<10:
    season=3
elif date<13:
    season=4
else:
    print("This would not work.")
print(season)