python - 如何让 get_dummies 工作到位？

Question

我在我的 DataFrame 上应用 get_dummies 来生成虚拟变量。它创建了一个新的 DataFrame。如何更改我原来的 DataFrame？

这行得通，但还有更好的方法吗？

import pandas as pd
data = pd.DataFrame({'gender': [ 'female', 'male']})
data1 = pd.get_dummies(data, columns = ['gender'])
# data is still unchanged
data.drop(data.columns, inplace=True, axis=1)
data[data1.columns] = data1

Answer 1

在您的代码中，您正在创建一个新数据框，然后从旧数据框中删除所有数据，然后将新数据放回旧数据框中。

除了最后三行代码，您可以只说:

data = pd.get_dummies(data, columns = ['gender'])

get_dummies 函数创建一个新数据框并将其保存在旧数据框的位置。这在功能上与您的代码相同，但更容易理解。