python - 我应该使用哪个语句来替换创建虚拟变量的自定义函数以提高 python 中的速度？

Question

我需要从现有的 python 数据帧变量创建几个虚拟变量。例如，列 x 的值从 1 到 3，我需要创建 x_dummy1，当 x=1 时，它的值为 1；否则为 0，x_dummy2 当 x=2 时取值为 1；否则为 0。目前我正在使用自定义函数:

def x_dummy(df):
    if df['x']==1:
        return 1
    else:
        return 0
df['x_dummy1']=df.apply(x_dummy, axis=1)

效果很好，但当文件很大时会花费很多时间。还有其他 python 语句可以用来绕过这个吗？谢谢。

Answer 1

只需创建带有零的虚拟列并根据条件替换为 1:

df["x_dummy1"] = 0
df.loc[df.x==1, "x_dummy1"] = 1