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)