数据框中的 2 个虚拟变量

Question

我有一个包含 5 个数值变量和 2 个分类变量的数据框 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)