Python：将计数器写入 csv 文件

我有一个包含列的 csv 数据文件‘number’, ’colour’, ’number2’, ’foo’, ’bar’，看起来像：

12, red, 124, a, 15p
14, blue, 353, c, 7g
12, blue, 125, d, 65h
12, red, 124, c, 12d

我想计算 number、color 和 number2 一起出现的次数，例如，上面列表的输出将是：’12, red, 124 :2’,’14, blue, 353: 1’, ’12, blue, 125: 1’。我已经使用以下方法完成了此操作：

import csv
datafile=open('myfile.csv','r')
usefuldata=[] 
for line in datafile: 
    usefuldata.append(line) 
from collections import Counter
outfile1=Counter((line[1],line[2],line[3]) for line in usefuldata)  
print(outfile1)

这给了我：

Counter({(‘12’,’red’,’135’): 21, (‘15’,’blue’,’152’):18, (‘34’,’green’,’123’):16 etc})

这很好，但我想把它写到一个文件中。我希望该文件有 4 列：数字、颜色、数字 2 和计数。我意识到这是一个常见问题，并且我尝试了其他线程上建议的几种不同方法，但没有一个有效。

Newfile=open(‘newfile.csv’,’wb’)
fieldnames=['a','b']
csvwriter=csv.DictWriter(newfile, delimiter=',', fieldnames=fieldnames)
csvwriter.writerow(dict((fn,fn) for fn in fieldnames))
for row in outfile1:
    csvwriter.writerow(row)

And

with open('newfile.csv','wb') as csvfile:
    fieldnames=['number','colour','number2']
    writer=csv.DictWriter(csvfile, fieldnames=fieldnames)
    writer.writeheader()
    writer.writerow(Counter((line[1],line[2],line[3]) for line in usefuldata))
    countwriter=csv.writer(csvfile, delimiter=', ')
    countwriter.writerow(outfile1)

两者都给我错误

    return self.writer.writerow(self._dict_to_list(rowdict))
TypeError: 'str' does not support the buffer interface

我也尝试过使用泡菜：

import pickle
with open('newfile.csv','wb') as outputfile:
    pickle.dump(outfile1, outputfile)

给我乱码文件。

我目前的尝试是使用

writer=csv.DictWriter(newfile, outfile1)
for line in outfile1:
    writer.writerow(line)

但这给了我一个关于字段名的错误。

我知道这是一个常见的问题，我意识到我只是在挣扎，因为我真的不知道我在做什么——自从我使用 python 以来已经有几年了，我已经忘记了很多。任何帮助将不胜感激。

首先说一下主要问题的原因——

TypeError: 'str' does not support the buffer interface

是您正在以二进制模式打开文件，您应该以文本模式打开文件（不b ).

其次，我想说使用普通的会更容易csv.writer https://docs.python.org/2/library/csv.html#csv.writer than csv.DictWriter() https://docs.python.org/2/library/csv.html#csv.DictWriter就您而言，因为您的字典的创建方式。

将结果写入 csv 的一种方法是 -

#Assuming you have previously created the counter you want to write
#lets say you stored the counter in a variable called cnter
with open('newfile.csv','w') as csvfile:
    fieldnames=['number','colour','number2','count']
    writer=csv.writer(csvfile)
    writer.writerow(fieldnames)
    for key, value in cnter.items():
        writer.writerow(list(key) + [value])