2017-02-20 16 views
0

我想從我的csv文件中提取列。爲此,我寫了一篇專欄文章,內容如下:如何在Pandas中按其名稱選擇列,並將該列作爲數組而不是字典

train_error,cv_error,test_error 
25.4965,25.3198,25.8298 
25.7468,25.4954,26.0417 
25.8253,25.6034,26.1515 
26.3315,26.0611,26.6243 
26.1648,25.92,26.4781 
26.2192,25.9808,26.5419 
26.3396,26.0814,26.6447 
26.2184,25.9735,26.5339 
26.2553,26.0105,26.5733 
26.2683,26.0142,26.5763 
26.1743,25.9286,26.4885 
26.1716,25.9236,26.4836 
26.1836,25.9345,26.495 
26.1548,25.9083,26.4684 
26.1505,25.9022,26.4617 
26.1342,25.887,26.4463 
26.1468,25.8994,26.4592 
26.1453,25.8976,26.4574 
26.1322,25.8854,26.445 
26.132,25.8842,26.4438 
26.1297,25.8828,26.4423 

大家都建議我用熊貓。我已經看到了它的API,並且我似乎無法以實際使用列名稱的方式獲取列。只有這樣,才能獲得,也就是說,train_errors0如下索引:

df = pd.read_csv('./tmp_errors/error_file.csv') 
print(df.as_matrix()[:,0]) 

那真的只有這樣,才能做到這一點?我希望我可以用我的代碼中的train_error這個字或者其他東西來使它更具可讀性,或者用這些鍵和數組來指向錯誤。如果我不能這樣做,我不得不使用0,1,2等索引,那麼甚至有命名列的點?

有沒有一種方法可以真正使用列的名稱來提取數據?


在對方的回答似乎是這樣做的:

df.to_dict() 

給我想要的,只是不用具有與該鍵作爲列名的字典幾乎什麼,該行是一本字典數字。如:

{'cv_error': {0: 25.319800000000001, 1: 25.4954, 2: 25.603400000000001, 3: 26.0611, 4: 25.920000000000002, 5: 25.980799999999999, 6: 26.081399999999999, 7: 25.973500000000001, 8: 26.0105, 9: 26.014199999999999, 10: 25.928599999999999, 11: 25.9236, 12: 25.9345, 13: 25.908300000000001, 14: 25.902200000000001, 15: 25.886999999999997, 16: 25.8994, 17: 25.897600000000001, 18: 25.885400000000001, 19: 25.8842, 20: 25.8828}, 'train_error': {0: 25.496500000000001, 1: 25.7468, 2: 25.825299999999999, 3: 26.331499999999998, 4: 26.1648, 5: 26.219200000000001, 6: 26.339600000000001, 7: 26.218399999999999, 8: 26.255299999999998, 9: 26.2683, 10: 26.174299999999999, 11: 26.171600000000002, 12: 26.183599999999998, 13: 26.154800000000002, 14: 26.150500000000001, 15: 26.1342, 16: 26.146799999999999, 17: 26.145299999999999, 18: 26.132200000000001, 19: 26.131999999999998, 20: 26.1297}, 'test_error': {0: 25.829799999999999, 1: 26.041699999999999, 2: 26.151499999999999, 3: 26.624300000000002, 4: 26.478100000000001, 5: 26.541899999999998, 6: 26.6447, 7: 26.533899999999999, 8: 26.5733, 9: 26.5763, 10: 26.488499999999998, 11: 26.483599999999999, 12: 26.495000000000001, 13: 26.468399999999999, 14: 26.4617, 15: 26.446300000000001, 16: 26.459199999999999, 17: 26.4574, 18: 26.445, 19: 26.4438, 20: 26.442299999999999}} 

我希望它會是這樣的:

{'train_error':[25.4965,...,26.1297]} 
換句話說

,充當陣列應該是一個數組不是愚蠢/怪異指數的字典詞典。

+0

'csv'模塊支持這樣'DictReader'所以有_is_一個點。 –

+0

[讀取csv文件作爲字典使用熊貓]可能的副本(http://stackoverflow.com/questions/33620982/reading-csv-file-as-dictionary-using-pandas) – Barmar

回答

4

選擇按列名是basic feature of pandas

df = pd.read_csv('./tmp_errors/error_file.csv') 
print df['train_error'] 
+0

但爲什麼我會得到一些奇怪的消息結束? 'Name:train_error,dtype:float64'你也明白嗎? – Pinocchio

+1

這是熊貓的內置功能,可以告訴您剛纔打印的系列的名稱和類型。這不是一個額外的行,如果您將其保存回CSV,它將不會成爲它的一部分。 – ASGM

+0

我在問題中增加了一個細節,你知道答案嗎? – Pinocchio

相關問題