Python中如何基于键将DataFrame与字典关联生成新列？

假设你的DataFrame里有一列用于标记成员所属团队的负责人（比如示例中Mike/Stella对应的Frank），我们可以借助Pandas的map()方法快速将字典中的团队成员数匹配到对应行，生成目标新列。

代码实现 #

假设DataFrame名为df，负责人列名为Team Lead：

import pandas as pd

# 你的团队成员数字典
team_dict = {'Frank': 2, 'Jake': 1, 'Rando': 1, 'Susan': 1, 'Steven': 1, 'Dan': 1}

# 添加新列
df['Number of Team Mates'] = df['Team Lead'].map(team_dict)

补充说明 #

• 如果DataFrame中存在字典未覆盖的负责人名称，map()会返回NaN，可以用fillna()补全：

df['Number of Team Mates'] = df['Team Lead'].map(team_dict).fillna(0)

• 若你的负责人列名称不是Team Lead，只需替换代码中对应的列名即可。

内容的提问来源于stack exchange，提问作者PyNub