如何让 get_dummies 正常工作？

Question

我应用get_dummies我DataFrame来生成虚拟变量。它创建了一个新的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 函数创建一个新的数据帧并将其保存在旧数据帧的位置。这在功能上与您的代码相同，但更容易理解。