使用 python 在 csv 文件中以特定方式添加列值

Question

我有一个类似于以下内容的 csv 文件：

title  title2  h1  h2  h3 ... 
l1.1     l1     1   1   0  
l1.2     l1     0   1   0
l1.3     l1     1   0   1
l2.1     l2     0   0   1
l2.2     l2     1   0   1
l3.1     l3     0   1   1
l3.2     l3     1   1   0
l3.3     l3     1   1   0
l3.4     l3     1   1   0    

我希望能够通过以下方式添加列：
h1 ( l1.1 + l1.2+ l1.3 ) = 2
h1 ( l2.1 + l2.2 ) = 1
h1 ( l3.1 + l3.2 + l3.3 +l3.4) = 3 每列依此类推 我想要每个这样的值的最终计数作为汇总表：

title2  h1  h2  h3...
l1     2   2   1
l2     1   0   2
l3     3   4   1

我如何实现这个？

Answer 1

像这样的东西应该有效。它接受表单中的输入

title,title2,h1,h2,h3
l1.1,l1,1,1,0
l1.2,l1,0,1,0
l1.3,l1,1,0,1
l2.1,l2,0,0,1
l2.2,l2,1,0,1
l3.1,l3,0,1,1
l3.2,l3,1,1,0
l3.3,l3,1,1,0
l3.4,l3,1,1,0

并输出

title2,h1,h2,h3
l1,2,2,1
l2,1,0,2
l3,3,4,1

Tested with Python 3.1.2。在 Python 2.x 中，您需要更改 open() 调用以使用二进制模式，并删除 newline="" 位。您还可以删除对 list() 的调用，因为在 Python 2.x 中，map() 已经返回一个列表。

import csv
import operator

reader = csv.reader(open("test.csv", newline=""), dialect="excel")
result = {}

for pos, entry in enumerate(reader):
    if pos == 0:
        headers = entry
    else:
        if entry[1] in result:
            result[entry[1]] = list(map(operator.add, result[entry[1]], [int(i) for i in entry[2:]]))
        else:
            result[entry[1]] = [int(i) for i in entry[2:]]

writer = csv.writer(open("output.txt", "w", newline=""), dialect="excel")
writer.writerow(headers[1:])

keys = sorted(result.keys())
for key in keys:
    output = [key]
    output.extend(result[key])
    writer.writerow(output)

Answer 2

看一下 csv 模块。您想要做的是使用 csv.reader 打开文件。然后，您迭代该文件，一次一行。您将添加的结果累积到临时列表中。完成后，您可以将此列表写入新的 csv.writer。

您可能需要定义一种方言，因为您实际上并没有使用 CSV，而是使用某种制表符分隔的格式。