Python数据预处理：类别数据转数值的三种常用方法

季宁

发布时间：2026-10-13 | 1人浏览过

原创

本文介绍Python中处理非数值类别数据的三种方法：使用LabelEncoder快速转换、通过字典映射手动指定数值，以及利用get_dummies生成虚拟变量。内容涵盖具体代码实现与适用场景，帮助开发者高效完成数据预处理。

在进行Python数据分析时，首要步骤通常是对数据进行预处理。

在实际操作中，我们常需处理非数值型的类别数据。本文将重点探讨针对这类数据的处理策略。

目前常见的处理手段主要包含以下三种：

第一种方法是利用LabelEncoder进行快速转换。

第二种方法是通过映射（mapping）方式，将特定类别手动映射为对应的数值。需要注意的是，该方法的适用范围相对有限。

第三种方法则是借助get_dummies函数进行转换。

import pandas as pd
from io import StringIO

csv_data = '''A,B,C,D
1,2,3,4
5,6,,8
0,11,12,'''

df = pd.read_csv(StringIO(csv_data))
print(df)
#统计为空的数目
print(df.isnull().sum())
print(df.values)

#丢弃空的
print(df.dropna())
print('after', df)
from sklearn.preprocessing import Imputer
# axis=0 列  axis = 1 行
imr = Imputer(missing_values='NaN', strategy='mean', axis=0)
imr.fit(df) # fit 构建得到数据
imputed_data = imr.transform(df.values) #transform 将数据进行填充
print(imputed_data)

df = pd.DataFrame([['green', 'M', 10.1, 'class1'],
          ['red', 'L', 13.5, 'class2'],
          ['blue', 'XL', 15.3, 'class1']])
df.columns =['color', 'size', 'price', 'classlabel']
print(df)

size_mapping = {'XL':3, 'L':2, 'M':1}
df['size'] = df['size'].map(size_mapping)
print(df)

## 遍历Series
for idx, label in enumerate(df['classlabel']):
  print(idx, label)

#1, 利用LabelEncoder类快速编码,但此时对color并不适合,
#看起来,好像是有大小的
from sklearn.preprocessing import LabelEncoder
class_le = LabelEncoder()
color_le = LabelEncoder()
df['classlabel'] = class_le.fit_transform(df['classlabel'].values)
#df['color'] = color_le.fit_transform(df['color'].values)
print(df)

#2, 映射字典将类标转换为整数
import numpy as np
class_mapping = {label: idx for idx, label in enumerate(np.unique(df['classlabel']))}
df['classlabel'] = df['classlabel'].map(class_mapping)
print('2,', df)


#3,处理1不适用的
#利用创建一个新的虚拟特征
from sklearn.preprocessing import OneHotEncoder
pf = pd.get_dummies(df[['color']])
df = pd.concat([df, pf], axis=1)
df.drop(['color'], axis=1, inplace=True)
print(df)

以上即为Python数据预处理中类别数据转数值方法的完整介绍，希望能为读者提供参考与支持。