python - Pandas 将 6 月 1 日至 7 月 30 日定义为季节性年份，而不是 1 月 1 日至 12 月 31 日

Question

我有季节性降雪数据，我想按降雪年份(1954 年 7 月 1 日 - 1955 年 6 月 30 日)分组，而不是将一个冬天的数据分成两年(1954 年 1 月 1 日 - 1954 年 12 月 31 日和 1 月 1 日) , 1955 - 1955 年 12 月 31 日。)

example data

我修改了这个问题的代码:

Using pandas to select specific seasons from a dataframe whose values are over a defined threshold (感谢垫)

def get_season(row):
  if row['date'].month <= 7:
      return row['date'].year
  else:
      return row['date'].year + 1

df['Seasonal_Year'] = df.apply(get_season, axis=1)

results of method call

有没有比我做的更好的方法？

Answer 1

我想是的，用 numpy.where :

years = df['date'].dt.year
df['Seasonal_Year'] = np.where(df['date'].dt.month <= 7, years, years + 1)