python - 为 Pandas 数据框创建新列的条件要求

Question

我有一个如下所示的数据框:

Name   IDNum      BossNum
John   1          0
Matt   2          1
Mike   3          2
Jack   4          1

如何创建一个名为 BossName 的附加列，如下所示:

预期输出:

Name   IDNum      BossNum     BossName
John   1          0           John/none (I don't care about boss being ided)
Matt   2          1           John
Mike   3          2           Matt
Jack   4          1           John

我目前已经尝试过:

df['BossName'] = df['Name'][df['BossNum'] in df['IDNum']]

这根本不起作用。关于我如何实现这一点的任何想法都会很棒。

Answer 1

您可以创建一个系列Name，索引为IDNum，然后使用map:

df['BossName'] = df.BossNum.map(df.set_index('IDNum').Name)
df