與大熊貓

Question

當加載一個CSV文件看起來像這樣

0 1 male 3 4 5 6 
1 0 female 6 7 8 9 
..... 

是可以自動轉換第三欄爲整數，例如0男1女分類變量？

Answer 1

read_csv接受名爲converters的參數。這可以作爲一個文件讀入到應用功能，以特定列converters應爲以下形式的字典中傳遞：

{column_index: function_to_apply} 

您可以使用此功能適用於第三列。所有你需要做的是設置函數來獲取從字典d它映射到"male"和0到"female"一個1值：

>>> d = {"male": 0, "female": 1} 
>>> pd.read_csv(file.csv, converters={2: d.get}) 
... 
0 1 0 3 4 5 6 
1 0 1 6 7 8 9 
...