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