迭代等周熊猫答案

【问题标题】：Iterating over Isoweeks pandas迭代等周熊猫
【发布时间】：2021-07-03 10:03:50
【问题描述】：

我有一个包含 date、value 和 isoweek 字段的数据框，如下所示：

date       | value | isoweek 
-----------------------------
2018-04-01 | 5     | 2018-13
2018-04-10 | 10    | 2018-15
2018-05-01 | 10    | 2018-18

其中isoweek 是对应日期的年-周。我的目标是遍历等周，找到数据中不存在的等周，并在数据框中插入一行，以0 作为值。

预期的输出如下所示：

date       | value | isoweek 
-----------------------------
2018-04-01 | 5     | 2018-13
NaN        | 0     | 2018-14
2018-04-10 | 10    | 2018-15
NaN        | 0     | 2018-16
NaN        | 0     | 2018-17
2018-05-01 | 10    | 2018-18

如何遍历原始数据框，并找到数据中所有缺失的等周？

【问题讨论】：

标签： python pandas date

【解决方案1】：

可能有点冗长，但您可以在将 isoweek 转换为日期后尝试使用resample：

s = pd.to_datetime(df['isoweek']+"-0",format='%Y-%W-%w')
u = df.set_index(s).resample("W").first()

iso_week = u.index.year.astype(str)+'-'+u.index.weekofyear.astype(str)
u['isoweek'] = u['isoweek'].fillna(pd.Series(iso_week,index=u.index))
out = u.fillna({"value":0}).reset_index(drop=True)

print(out)

         date  value  isoweek
0  2018-04-01    5.0  2018-13
1         NaN    0.0  2018-14
2  2018-04-10   10.0  2018-15
3         NaN    0.0  2018-16
4         NaN    0.0  2018-17
5  2018-05-01   10.0  2018-18

【讨论】：

【解决方案2】：

你可以试试apply:

def func(row):
    year = (row.name)
    r = row['isoweek'].str.split('-').str[1].astype(int)
    min_week = min(r)
    max_week = max(r)
    val_range = range(min_week, max_week)
    missing = (set(val_range) - set(r.values))
    for mis_week in missing:
        row = (row.append({'isoweek': f"{year}-{mis_week}", 'date': np.nan, 'value':0}, ignore_index=True))
    return (row.sort_values(by='isoweek').reset_index(drop=True))

df.groupby(df.isoweek.str.split('-').str[0],as_index=False).apply(func)

        date        value   isoweek
 0  0   2018-04-01  5       2018-13
    1   NaN         0       2018-14
    2   2018-04-10  10      2018-15
    3   NaN         0       2018-16
    4   NaN         0       2018-17
    5   2018-05-01  10      2018-18

【讨论】：

【解决方案3】：

您可以使用pandas.date_range() 生成每周从start 到end 的日期列表。

dates = pd.Series(pd.date_range(start=df['date'].min(), end=df['date'].max(), freq='W'))
isoweeks = (dates.dt.isocalendar().year.astype(str) + '-' + dates.dt.isocalendar().week.astype(str)).tolist()
max_isoweek = str(df['date'].max().isocalendar()[0]) + '-' + str(df['date'].max().isocalendar()[1])

if max_isoweek not in isoweeks:
    isoweeks.append(max_isoweek)

这是为了获取 start 日期和 end 日期之间的所有 iso 周。

然后您可以将 df 合并到一个帮助数据帧中以获得您想要的。

df = df.merge(pd.DataFrame({'isoweek': isoweeks}), how='right')
df['value'].fillna(0, inplace=True)

# print(df)

        date  value  isoweek
0 2018-04-01    5.0  2018-13
1        NaT    0.0  2018-14
2 2018-04-10   10.0  2018-15
3        NaT    0.0  2018-16
4        NaT    0.0  2018-17
5 2018-05-01   10.0  2018-18

【讨论】：