列表中每个元素的新列

Question

我有一个有很多名字的数据集。我想为每个特定名称创建一个新列，如果名称相同则为 1，否则为 0。

原始数据：

期望的输出：

我尝试了以下方法：

names=['Tom','Sarah','Bob']

def function(x):
    for n in names:
        if (x['Name']==n):
            return 1
        else:
            return 0
        
for n in names:        
    df[n]=df.apply(function,axis=1)

这不起作用，因为它返回所有名称的“Tom”列：

我做错了什么？

Answer 1

你可以get_dummies

out = df.join(df.Name.str.get_dummies()[names])