根据包含列总和的列将 CSV 文件分成多个文件

Question

我在 stackoverflow https://stackoverflow.com/questions/31081953/split-csv-file-into-two-csv-files-based-on-the-content-of-a-column 上找到了一个解决方案，用于将文件分成多个列，如下所示

import csv
from itertools import groupby

for key, rows in groupby(csv.reader(open("newone.csv")),
                         lambda row: row[0]):
    with open("%s.csv" % key, "w") as output:
        for row in rows:
            output.write(",".join(row) + "\n")

这可以很好地完成工作，但我想打印出某些列中的数字总和，例如第 3 列和第 4 列，我该怎么做？
看看这张图片 - http://i.imgur.com/KIxLOnr.jpg 此代码工作正常，但我需要在该图像中生成以橙色突出显示的最后一行（我手动添加了那些）。我该怎么做？

Answer 1

这将在每行的末尾添加数字：

for row in rows:
    output.write(",".join(row))
    sum_of_certain_cols = row[2] + row[3]
    output.write("," + sum_of_certain_cols + "\n")

如果你想把总和放在某个位置，你应该做一个insert，比如：

    sum_of_certain_cols = row[2] + row[3]
    row.insert(4, str(sum_of_certain_cols))
    output.write(",".join(row) + "\n")

在 cmets 之后编辑

1. 不要忘记将列转换为数据类型，尽管我认为不再需要此行，因为我了解您的需求：

   sum_of_certain_cols = int(row[2]) + int(row[3])
   

2. 如果你想在最后加总，你需要一个累加器，例如：

   sums = [0] * len(rows)
   

   然后在循环内做所有的和：

   sums[1] += int(row[1])
   sums[2] += int(row[2])
   

   然后在循环之后把它们都写出来：

   output.write(",".join(sum) + "\n")