使用某些字段查找重复记录并使用 Pandas 库更改字段值

Question

Week_number   Holiday Description  Qty
38              1        A          5
38              0        A          6
38              0        B          1
38              1        C          1
40              0        A          1

我想查找相同 Week_number 和 Description 的重复项。如果我们以上述 Week_number 38 和 Desciption A 为例，则有 2 条记录。然后我想得到这 2 个数量的总和，所以它将是 11。 最后合并这两条记录，将 Qty 和 Holiday 的总和显示为 1。

Week_number   Holiday Description  Qty
38              1        A          11
38              0        B          1
38              1        C          1
40              0        A          1

再次检查 Week_number 38 的重复项，将有 3 条记录。并将 Holiday 更改为所有具有相同 Week_number 的记录中的 1 个。

Week_number   Holiday Description  Qty
38              1        A          11
38              1        B          1
38              1        C          1
40              0        A          1

任何cmets如何做到这一点？ 谢谢

Answer 1

查找week_number，其中holiday == 1。然后通过将Holiday 分配给该特定周数的1，将剩余的0 转换为1。最后一部分是根据子集 = ['Week_number','Description']

删除重复项

df['Qty'] = df.groupby(['Description','Week_number']).Qty.transform('sum')

cond = df.query('Holiday ==1').Week_number.unique()

df['Holiday'] = np.where(df.Week_number.isin(cond),1,df.Holiday)

df = df.drop_duplicates(['Week_number','Description'])

    Week_number Holiday Description  Qty
0       38          1       A        11
2       38          1       B         1
3       38          1       C         1
4       40          0       A         1