python - 如何合并行和添加列值？

Question

所以我有一个像这样的 laaaaaaaarge 文件:

Item|Cost1|Cost2
Pizza|50|25
Sugar|100|100
Spices|100|200
Pizza|100|25
Sugar|200|100
Pizza|50|100

我想为特定项目添加所有 Cost1 和 Cost2 并生成合并输出。

我已经编写了一个 python 代码来执行此操作，

item_dict = {}
for line in file:
    fields = line.split('|')
    item = fields[0]
    cost1 = fields[1]
    cost2 = fields[2]
    if item_dict.has_key(item):
        item_dict[item][0] += int(cost1)
        item_dict[item][1] += int(cost2)
    else:
        item_dict[item] = [int(cost1),int(cost2)]

for key, val in item_dict.items():
    print key,"|".join(val)

有没有办法在 awk 或使用任何其他魔法中非常有效和快速地做到这一点？

或者我可以让我的 python 更优雅、更快吗？

预期输出

Pizza|200|150
Sugar|300|200
Spices|100|200

Answer 1

像这样的……

$ awk 'BEGIN{OFS=FS="|"}
  NR>1 {cost1[$1]+=$2; cost2[$1]+=$3} 
  END{ for (i in cost1) print i, cost1[i], cost2[i]}' file
Sugar|300|200
Spices|100|200
Pizza|200|150

说明

• BEGIN{OFS=FS="|"} 将(输入和输出)字段分隔符设置为 |。
• NR>1 表示我们将对大于 1 的行号执行一些操作。这样我们就可以跳过标题。
• cost1 和 cost2 是数组，其索引是第一个字段，其值是到该点的总和。
• END {} 是我们在读取整个文件后所做的事情。它包括遍历数组并打印值。