Python 根据多个条件将 dataframe 的行添加/合并在一起

Question

下午好，希望你一切都好。

我有一个以下格式的 xlsx 文件，它是 output 来自 Python function 我一直用来解析数据：-

我已将此 xlsx 文件加载到 pandas df 中，以尝试实现以下 output：-

我要满足的要求是：- 对于 dataframe 中的每一行，如果“应用程序 ID”和“测试阶段”列值匹配，那么我想将这些列值的行值添加在一起并替换原始匹配行，其中一行包含总和值。

如果列值匹配，则原始行应保留在原位。

如果有任何关于如何实现这一目标的指示，将不胜感激。 I have attempted to achieve this code in the function prior to writing the values to the source xlsx output file however I assumed it would be easier to achieve by working with pandas / numpy.

非常感谢吉米

Answer 1

使用groupby_sum ：

out = df.groupby(['Application ID', 'Test Phase'], as_index=False).sum()
print(out)

# Output
   Application ID Test Phase  Total Tests   A
0               9        SIT           36  36
1              11        UAT            5   5

设置：

data = {'Application ID': [9, 9, 11],
        'Test Phase': ['SIT', 'SIT', 'UAT'],
        'Total Tests': [9, 27, 5],
        'A': [9, 27, 5]}
df = pd.DataFrame(data)
print(df)

# Output
   Application ID Test Phase  Total Tests   A
0               9        SIT            9   9
1               9        SIT           27  27
2              11        UAT            5   5