2011-09-26 94 views
0

我需要知道如何總結CSV文件中列的所有數字。如何總結python中一列的所有數字?

例如。我的數據是這樣的:

column count min max sum mean 
80 29573061 2 40 855179253 28.92 
81 28861459 2 40 802912711 27.82 
82 28165830 2 40 778234605 27.63 
83 27479902 2 40 754170015 27.44 
84 26800815 2 40 729443846 27.22 
85 26127825 2 40 701704155 26.86 
86 25473985 2 40 641663075 25.19 
87 24827383 2 40 621981569 25.05 
88 24189811 2 40 602566423 24.91 
89 23566656 2 40 579432094 24.59 
90 22975910 2 40 553092863 24.07 
91 22412345 2 40 492993262 22 
92 21864206 2 40 475135290 21.73 
93 21377772 2 40 461532152 21.59 
94 20968958 2 40 443921856 21.17 
95 20593463 2 40 424887468 20.63 
96 20329969 2 40 364319592 17.92 
97 20157643 2 40 354989240 17.61 
98 20104046 2 40 349594631 17.39 
99 20103866 2 40 342152213 17.02 
100 20103866 2 40 335379448 16.6 
#But it's separated by tabs 

到目前爲止,我編寫的代碼是:

import sys 
import csv 

def ErrorCalculator(file): 
     reader = csv.reader(open(file), dialect='excel-tab') 

     for row in reader: 
       PxCount = 10**(-float(row[5])/10)*float(row[1]) 


if __name__ == '__main__': 
     ErrorCalculator(sys.argv[1]) 

對於這個特殊的代碼,我需要所有的總和,總結在PxCount所有的數字和鴻溝行號[1]中的數字...

如果告訴我如何總結列的數字或者如果您幫助我使用此代碼,我將非常感激。

此外,如果你可以給我一個提示跳過標題。

回答

2

你可以叫「讀卡器。 next()「在實例化讀取器後放棄第一行。

要計算PxCount,只需在循環前設置sum = 0,然後在爲每行計算後設置sum += PxCount

PS您可能會發現csv.DictReader也有幫助。

2

你可以使用"augmented assignment" +=保持運行總計:

total=0 
for row in reader: 
     PxCount = 10**(-float(row[5])/10)*float(row[1]) 
     total+=PxCount 

要跳過CSV文件的第一行(標題):

with open(file) as f: 
    next(f) # read and throw away first line in f 
    reader = csv.reader(f, dialect='excel-tab') 
1

使用DictReader將產生更清晰的代碼。 Decimal會給你更好的精度。另外嘗試遵循Python命名約定,併爲函數和變量使用小寫名稱。

import decimal 

def calculate(file): 
    reader = csv.DictReader(open(file), dialect='excel-tab') 
    total_count = 0 
    total_sum = 0 
    for row in reader: 
     r_count = decimal.Decimal(row['count']) 
     r_sum = decimal.Decimal(row['sum']) 
     r_mean = decimal.Decimal(row['mean']) 
     # not sure if the below formula is actually what you want 
     total_count += 10 ** (-r_mean/10) * r_count 
     total_sum += r_sum 
    return total_count/total_sum 
相關問題