熊貓get_dummies在for循環

Question

我想在for循環中使用pandas.get_dummies將類別變量轉換為虛擬變量。 但是，以下代碼不會轉換數據幀。

data_cleaner = [data_train, data_val]
for df in data_cleaner:
    df = pd.get_dummies(df, columns = categorical_fields)

data_train.head() # Not changed

我知道for循環中的迭代器只是一個臨時變量。 但是修改后的代碼也不起作用。

for i in range(len(data_cleaner)):
    data_cleaner[i] = pd.get_dummies(data_cleaner[i], columns = categorical_fields)

data_train.head() # Still not changed

有人可以幫忙嗎？ 我是否必須為每個數據幀手動運行get_dummies ？ 僅供參考，熊貓get_dummies不提供inplace選項。

Answer 1

嘗試跟隨

data_cleaner = [data_train, data_val]
for i,df in enumerate(data_cleaner):
    data_cleaner[i] = pd.get_dummies(df, columns = categorical_fields)

data_train,data_val=data_cleaner

Answer 2

您可以將其作為列表理解

data_cleaner = [pd.get_dummies(df, columns=categorical_fields) for df in data_cleaner]

要么

data_train_dum, data_val_dum = [pd.get_dummies(df, columns=categorical_fields) for df in [data_train, data_val]]