dataframe 中的 2 个虚拟变量

Question

我有 5 个数值变量和 2 个分类变量的 dataframe x_0：['v1'，'v2'，'v3'，'v4'，'v5'，'v6'，'v7']

我想将 v6 和 v7 转换为虚拟变量。 我已尝试使用以下代码，但收到回溯：

x = pd.concat([x_0['v1', 'v2', 'v3', 'v4', 'v5'], pd.get_dummies(x_0['v6', 'v7'])], axis = 1)

Answer 1

使用 double []选择多个列名：

x = pd.concat([x_0[['v1', 'v2', 'v3', 'v4', 'v5']], 
               pd.get_dummies(x_0[['v6', 'v7']])], axis = 1)