根据另一个 Date 列在 Pandas 数据帧上生成周数

Question

我正在使用以下代码生成如下数据框：

我面临的挑战是周数将在 53.0 结束，然后重置回 1.0，之后周数继续计算。

我需要一个解决方案，其中周数持续到 53.0、54.0 直到日期结束（在第一列中找到）。

import pandas as pd
s = pd.date_range('2006-12-16', '2010-11-26', freq='D').to_series()
s.dt.dayofweek

sample = s.dt.dayofweek
df_sample=sample.to_frame()
df_sample.rename(columns = {0: 'Day_num'}, inplace=True)

df_sample.loc[df_sample.Day_num == 5, 'Day_name'] = 'Saturday'
df_sample.loc[df_sample.Day_num == 6, 'Day_name'] = 'Sunday'
df_sample.loc[df_sample.Day_num == 0, 'Day_name'] = 'Monday'
df_sample.loc[df_sample.Day_num == 1, 'Day_name'] = 'Tuesday'
df_sample.loc[df_sample.Day_num == 2, 'Day_name'] = 'Wednesday'
df_sample.loc[df_sample.Day_num == 3, 'Day_name'] = 'Thursday'
df_sample.loc[df_sample.Day_num == 4, 'Day_name'] = 'Friday'

df_sample = df_sample['Day_name'].astype(str)
df_sample = df_sample.to_frame()

df_sample.reset_index(level=0, inplace=True)
df_sample.rename(columns={'index': 'Date'}, inplace=True)

df_sample['Week_Number'] = df_sample['Date'].dt.week

我觉得有一个更简单的解决方案，并请求有更好方法解决这个问题的人提供任何帮助。

Answer 1

你可以使用：

s = pd.date_range('2006-12-16', '2010-11-26', freq='D')
df_sample=s.to_frame()
df_sample['Day_Name']=s.day_name()
df_sample=df_sample.reset_index(drop=True).rename(columns={0:'Date'})
week_range=np.arange(df_sample.index[0],df_sample.index[len(df_sample.index)-1],7)
df_sample['Week_Number']=df_sample.groupby(pd.cut(df_sample.index,week_range,right=False)).ngroup(ascending=False).sort_values().reset_index(drop=True)+1