Python：读取和写入CSV文件

Question

我试图从CSV文件（A）中读取数据，提取数据并将其写入不同的CSV文件（B）。在新的文件B中，我想要有两列。列1列出文件A和列2中列1的名称以列出文件A中列1的计数。例如，如果文件A看起来像这样没有'：'（它们排成两列）：

Animal: Gender 
Rabbit: Male 
Dog: Male 
Rabbit: Female 
Cat: Male 
Cat: Male 
Dog: Female 
Dog: Male 
Turtle: Male 

我想在文件B输出到这个样子（实际上是没有不同的列“：”再一次）：

Animal: Count 
Cat: 2 
Dog: 3 
Rabbit: 2 
Turtle: 1 

这是第一次我做这样的事，并这是我目前为止的内容，但是我没有将数据打印在文件B中，并且“计数”正确完成。有谁能帮我解决这个问题吗？

import csv 
ReadData=csv.reader(open('C:\Users\..\FileA.csv','rb'), delimiter=',') 

def column(ReadData, i): 
    return [row[i] for row in ReadData] 

for line in ReadData: 
    WriteData=csv.writer(open('C:\Users\..\FileB.csv','wb'), 
         delimiter=' ', quotechar=':', quoting=csv.QUOTE_ALL) 
    print column(ReadData,1) 

非常感谢您的帮助！

Answer 1

我会回答你问题的计数部分，也许你可以把它与你的问题的csv部分结合起来。

l = [ 
    ('Animal','Gender'), 
    ('Rabbit','Male'), 
    ('Dog','Male'), 
    ('Rabbit','Female'), 
    ('Cat','Male'), 
    ('Cat','Male'), 
    ('Dog','Female'), 
    ('Dog','Male'), 
    ('Turtle','Male') 
    ] 

d = {} 
for k,v in l: 
    if not k in d: 
     d[k] = 1 
    else: 
     d[k] += 1 

for k in d: 
    print "%s: %d" % (k,d[k]) 

我没有筛选标题行，这段代码的输出是：

Turtle: 1 
Cat: 2 
Rabbit: 2 
Animal: 1 
Dog: 3 

编辑：

您可以替换此：

if not k in d: 
    d[k] = 1 
else: 
    d[k] += 1 

有了这个：

d[k] = d.setdefault(k,0) + 1 

Answer 2

要在Python> = 2.7中进行计数，请参阅this example for collections.Counter。使用collections.defaultdict，请参见here。

在您拨打csv.writer时，quotechar=':'可能是一个错误（这将使得WriteData.writerow(['Hello World', 12345]）发出“：Hello World：12345”，就好像冒号是引号。

另请注意，您的功能column(ReadData, i)消耗ReadData;随后对ReadData的调用可能会返回一个空列表（未测试）。这对你的代码来说不是问题（至少现在不是）。

这是没有的CSV模块的解决方案（毕竟，这些文件不看太像CSV）：

import collections 

inputfile = file("A") 

counts = collections.Counter() 

for line in inputfile: 
    animal = line.split(':')[0] 
    counts[animal] += 1 

for animal, count in counts.iteritems(): 
    print '%s: %s' % (animal, count) 

Answer 3

看一看的itertools模块和groupby功能。例如：

from itertools import groupby 

animals = [ 
    ('Rabbit', 'Male'), 
    ('Dog', 'Male'), 
    ('Rabbit', 'Female'), 
    ('Cat', 'Male'), 
    ('Cat', 'Male'), 
    ('Dog', 'Female'), 
    ('Dog', 'Male'), 
    ('Turtle', 'Male') 
    ] 

def get_group_key(animal_data): 
    return animal_data[0] 

animals = sorted(animals, key=get_group_key) 
animal_groups = groupby(animals, get_group_key) 

grouped_animals = [] 
for animal_type in animal_groups: 
    grouped_animals.append((animal_type[0], len(list(animal_type[1])))) 

print grouped_animals 

>>> [('Cat', 2), ('Dog', 3), ('Rabbit', 2), ('Turtle', 1)] 

Answer 4

根据数据和复杂的大小...你可能要考虑使用pandas - 在http://pandas.pydata.org/信息和可用PyPI上。

但是请注意，这可能是过度杀戮，但我认为我会把它扔到混合。

from pandas import DataFrame 

# rows is processed from string in the OP 
rows = [['Rabbit', ' Male'], ['Dog', ' Male'], ['Rabbit', ' Female'], ['Cat', ' Male'], ['Cat', ' Male'], ['Dog', ' Female'], ['Dog', ' Male'], ['Turtle', ' Male']] 

df = pandas.DataFrame(rows, columns=['animal', 'gender']) 

>>> df.groupby('animal').agg(len) 
     gender 
animal   
Cat   2 
Dog   3 
Rabbit  2 
Turtle  1 

>>> df.groupby(['animal', 'gender']).agg(len) 
animal gender 
Cat  Male  2 
Dog  Female 1 
     Male  2 
Rabbit Female 1 
     Male  1 
Turtle Male  1