pandas基于另一个选定的列valu创建新列

Question

我有一个列user_id具有不同的用户id。 从那里我想选择特定的用户id并创建另一列来标识这些选定的用户id属于哪个组。在

例如，用户id“attamm1，attbmm1”属于与用户id匹配的另一个新创建列的“Vendor ABC”。在

我怎么能做到呢，谢谢！在

#map user id
userid_list = ['attamm1', 'attbmm1']
df['Vendor ABC'] = df[df.STRUCTURALSTATUS.isin(value_list)]

输出：

expected out screenshot

Answer 1

你可以用.loc

df.loc[df.STRUCTURALSTATUS.isin(value_list), 'Vendor ABC'] = df.loc[df.STRUCTURALSTATUS.isin(value_list), 'user_id']