如何创建新列并按列组分配值

【问题标题】：How to create new column and assign values by column group如何创建新列并按列组分配值
【发布时间】：2023-02-06 21:30:36
【问题描述】：

我有数据框，

df
uid    
 1     
 2      
 3
 ...

我想分配一个新列，值 0 或 1 取决于我将分配的 uid。

df
uid  new 
1     0
2     0
3     1
..

【问题讨论】：

【解决方案1】：

您必须解释底层逻辑。

也就是说，有很多可能的方法。

考虑与 map 的显式映射：

mapper = {1: 0, 2: 0, 3: 1}

df['new'] = df['uid'].map(mapper)

或者使用 1 和 isin 的目标列表并转换为 int

target = [3]

df['new'] = df['uid'].isin(target).astype(int)

输出：

   uid  new
0    1    0
1    2    0
2    3    1

【讨论】：