Pandas 按选定日期分组

Question

我有一个与此数据框非常相似的数据框：

索引	日期	月份
0	2019-12-1	12
1	2020-03-1	3
2	2020-07-1	7
3	2021-02-1	2
4	2021-09-1	9

我想合并最接近一组月份的所有日期。月份需要像这样标准化：

月份	标准化月份
3, 4, 5	4
6, 7, 8, 9	8
1, 2, 10, 11, 12	12

因此输出将是：

索引	日期	月份
0	2019-12- 1	12
1	2020-04-1	4
2	2020-08-1	8
3	2020-12-1	12
4	2021-08-1	8

Answer 1

您可以迭代 DataFrame 并使用替换来更改日期。

import pandas as pd 

df = pd.DataFrame(data={'date': ["2019-12-1", "2020-03-1", "2020-07-1", "2021-02-1", "2021-09-1"], 
                        'month': [12,3,7,2,9]})
for index, row in df.iterrows():
    if (row['month'] in [3,4,5]):
        df['month'][index] = 4
        df["date"][index]  = df["date"][0].replace(df["date"][0][5:7],"04")
    elif (row['month'] in [6,7,8,9]):
        df['month'][index] = 8
        df["date"][index]  = df["date"][0].replace(df["date"][0][5:7],"08")
    else:
        df['month'][index] = 12
        df["date"][index]  = df["date"][0].replace(df["date"][0][5:7],"12")
    

Answer 2

您可以尝试创建月份字典，其中：

norm_month_dict = {3: 4, 4: 4, 5: 4, 6: 8, 7: 8, 8: 8, 9: 8, 1: 12, 2: 12, 10: 12, 11: 12, 12: 12}

然后使用此字典将月份值映射到各自的标准化月份价值观。

df['normalized_months'] = df.months.map(norm_month_dict)

Answer 3

您需要从第二个数据帧构造一个字典（假设df1和df2）：

d = (
 df2.assign(Months=df2['Months'].str.split(', '))
    .explode('Months').astype(int)
    .set_index('Months')['Normalized month'].to_dict()
)
# {3: 4, 4: 4, 5: 4, 6: 8, 7: 8, 8: 8, 9: 8, 1: 12, 2: 12, 10: 12, 11: 12, 12: 12}

然后映射值：

df1['month'] = df1['month'].map(d)

输出：

   index        date   month
0       0  2019-12-1      12
1       1  2020-03-1       4
2       2  2020-07-1       8
3       3  2021-02-1      12
4       4  2021-09-1       8`