在製表符分隔文件上執行計算

Question

我仍在學習Python，但我仍然不知道如何使用數組。

我想讀一個製表符分隔文件，例如（但對我來說，我將有大約400行）：

Col1  Col2 
    0.0001 0.6 
    0.0001 0.5 
    0.000006 0.8 
    0.0001 0.0003 
    0.002  1 
    0.002  3 

我想要得到以下的輸出：

Col1  Col2 
0.0001 0.36676667 
0.000006 0.8 
0.002  2 

所以我想保持相同的值Col1中但考慮值的平均值在col2的這對應於Col1中相同的值。

我可以用讀數組：

arr = np.genfromtxt('test.csv', dtype=None, delimiter='\t', skiprows=1) 

，但我不知道如何進行這些操作，使新產生的數據的新文件。

非常感謝您的幫助！

Answer 1

使用collections.defaultdict並將list作爲默認參數。

將第一列的值作爲關鍵字，並追加第二個值。

import csv 
from collections import defaultdict 

# Gather the data from the CSV file 
d = defaultdict(list) 
with open('data.csv', 'r') as csvfile: 
    reader = csv.reader(csvfile, delimiter='\t') 
    for row in reader: 
     d[float(row[0])].append(float(row[1])) 

# Print the mean. 
for k in d.keys(): 
    print k, sum(d[k])/len(d[k])