2017-10-06 43 views
-1

我有一個數據集,如education{primary,graduate},martial status{male,female},job{employed, service,unemployed}。這是分類數據集,我想將它轉換爲浮點數,用於在numpy python中進行邏輯迴歸。我有一個分類numpy數組,想將其轉換爲整數?

+0

小記:我不認爲「服務」和「失業」是互斥的。 – tar

+0

它是一個例子。問題是我想將所有轉換爲float/Int數據類型。 –

+0

到目前爲止您嘗試了什麼? –

回答

1

一個選項沃爾德要創建字典和類似應用它在列表理解:

education = {'primary':0, 'graduate':1} 
ed_data = ['primary', 'primary', 'graduate', 'primary', 'graduate'] 

num_ed_data = [educaiton[i] for i in ed_data] 
# Output: [0, 0, 1, 0, 1]