python - 将具有多个键的Python字典映射到具有多列匹配键的数据框

Question

我有一个字典，我想将其映射到当前数据框并创建一个新列。我在一个元组中有键，它们映射到我的数据框中的两个不同列。

dct = {('County', 'State'):'CountyType'}
df = pd.DataFrame(data=['County','State'])

我想创建一个新列 CountyType，使用 dict 映射到 df 中的两列。但是，执行以下操作会给我一个错误。还有什么办法可以做到这一点？

df['CountyType'] = (list(zip(df.County,df.State)))
df = df.replace({'CountyType': county_type_dict)

Answer 1

您可以从两个系列创建一个MultiIndex，然后映射。来自@ALollz 的数据。

df['CountyType'] = df.set_index(['County', 'State']).index.map(dct.get)

print(df)

  County  State CountyType
0      A      1        One
1      A      2       None
2      B      1       None
3      B      2        Two
4      B      3      Three