我有一个分类的numpy数组，想把它转换成整数

Question

我有一个数据集，如education{primary,graduate}、martial status{male,female}、job{employed, service,unemployed}。这是分类数据集，我想将其转换为浮点数，以便在 numpy python 中进行逻辑回归。

Answer 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]