如何在 python pandas 中使用字典添加列

Question

我想在数据框中使用字典键添加一列 这是我尝试的当前代码，但它没有返回行政区而是为所有人返回“其他”

boroughs = {'Manhattan':[1, 2, 3], 
            'Bronx':[4, 5, 6], 
            'Brooklyn':[7, 8, 9],
            'Staten Island': 10}

def test(x):
  for key, value in boroughs.items():
    if int(x) in value or int(x) == value:
      return key
    else:
      return 'Other'
      
df['Boroughs'] = df.precinct.apply(test)

我运行代码后，这是当前的自治市镇

precinct   Boroughs
1.0        'Other'
5.0        'Other'
9.0        'Other'
10.0       'Other'

这是预期的结果

precinct   Boroughs
1.0        'Manhattan'
5.0        'Bronx'
9.0        'Brooklyn'
10.0       'Staten Island'

我想知道我哪里弄错了提前谢谢

Answer 1

你的 dict 是倒退的。试试这个：

boroughs = {
    'Manhattan': [1, 2, 3], 
    'Bronx': [4, 5, 6], 
    'Brooklyn': [7, 8, 9],
    'Staten Island': [10],
}

d = {}
for k, v in boroughs.items():
    for num in v:
        d[num] = k

df["Boroughs"] = df.precinct.astype(int).map(d)

输出：

   precinct       Boroughs
0       1.0      Manhattan
1       5.0          Bronx
2       9.0       Brooklyn
3      10.0  Staten Island

旁注：不保证 dicts 是可逆的。在这里，我假设每个区域都与一个独特的自治市镇相关联。