在Pandas中为Python创建虚拟变量

Question

我正在尝试使用python中的pandas从类别变量中创建一系列虚拟变量。我遇到过该get_dummies函数，但是每当我尝试调用该函数时，都会收到一个错误，提示您未定义名称。

创建虚拟变量的任何想法或其他方式将不胜感激。

编辑 ：由于其他人似乎遇到了这种情况，因此get_dummies熊猫中的函数现在可以正常工作了。这意味着以下应该起作用：

import pandas as pd

dummies = pd.get_dummies(df['Category'])

有关更多信息，请参见http://blog.yhathq.com/posts/logistic-regression-and-
python.html
。

Answer 1

很难从问题中推断出您正在寻找什么，但我的最佳猜测如下。

如果我们假设您有一个DataFrame，其中某些列为“ Category”（类别）并且包含类别的整数（或其他唯一标识符），那么我们可以执行以下操作。

调用DataFrame dfrm，并假设对于每一行，它dfrm['Category']是1到N之间的整数集中的某个值。然后，

for elem in dfrm['Category'].unique():
    dfrm[str(elem)] = dfrm['Category'] == elem

现在，根据该行中的数据是否在该类别中，每个类别都有一个新的指示符列，该列为是/否。

如果要控制类别名称，可以制作一个字典，例如

cat_names = {1:'Some_Treatment', 2:'Full_Treatment', 3:'Control'}
for elem in dfrm['Category'].unique():
    dfrm[cat_names[elem]] = dfrm['Category'] == elem

导致具有指定名称的列，而不仅仅是类别值的字符串转换。实际上，对于某些类型，str()可能不会产生任何对您有用的东西。