虚拟变量

在本课程的最后一个练习中，您将通过对非数值列进行虚拟编码来为建模做数据准备。 例如，若某一列为性别，取值为 'Male' 和 'Female'，您需要将其拆分为多个新列，用以指示该观测是否为 'Male' 或 'Female'。创建虚拟变量的过程也称为 one-hot 编码。

您可以使用 pandas 的 get_dummies() 函数将非数值列转换为虚拟变量。

df_new = pd.get_dummies(df)

我们已将 flights 数据帧取子集为 flights_sub，以便更清楚地看到每一步发生了什么。