python - 如何根据数据框中另一列的值将数据输入到新列中？

Question

我有一个非常大的数据框架，其中包含竞选财务数据 - 该数据框架有一列用于候选人委员会名称，但没有一列用于候选人名称。

如何根据委员会名称自动为候选人姓名添加新列？我希望代码读取每一行中的委员会名称，然后分配适当的候选人名称。

Answer 1

假设委员会名称和候选人名称之间存在一对一的映射(您可以在字典中描述)，您只需要使用 map功能。

示例代码:

>>> import pandas as pd
>>> data  = pd.DataFrame([["A"], ["B"], ["A"], ["C"]], columns=["Committee Name"])
>>> data
  Committee Name
0              A
1              B
2              A
3              C
>>> committee_name_mapping = {"A": "AName", "B": "BName", "C": "CName"}
>>> data["Candidate Name"] = data["Committee Name"].map(committee_name_mapping)
>>> data
  Committee Name Candidate Name
0              A          AName
1              B          BName
2              A          AName
3              C          CName