如何使用嵌套字典替换其他列中的字符串

Question

我有一个像下面这样的嵌套字典，如果该键位于字符串的末尾，我想使用内部键值对替换字符串，仅当国家代码等于 dict 键（不是内部键）时才用值替换它

'IND': {' PVT. LTD.': ' Pvt. Ltd.',
 ' pvt. Ltd': ' Pvt. Ltd.',
 ' PVT LTD': ' Pvt. Ltd.',
 ' L.L.P.': ' LLP',
 ' LTD.': ' Ltd.',
 ' LLP.': ' LLP',
 ' ltd': ' Ltd.',
 ' llp': ' LLP'},
 'GBR': {' P.L.C.': ' PLC',
 ' C.I.C.': ' CIC',
 ' p.l.c': ' PLC',
 ' c.i.c': ' CIC',
 ' s.e.': ' SE',
 ' PLC.': ' PLC'},
 'USA': {' LTD. CO.': ' Ltd. Co.',
 ' L.L.L.P.': ' LLLP',
 ' ltd. Co': ' Ltd. Co.',
 ' l.l.l.p': ' LLLP',
 ' L.L.P.': ' LLP',
 ' L.L.C.': ' LLC',
 ' l.l.p': ' LLP',
 ' l.l.c': ' LLC'}

我的数据框有两个列。法定名称和注册国家代码 -

Name	Reg Country Code
NexPoint LTD. CO.	USA
Silverplay P.L.C.	GBR
ALLOYS PVT. LTD.	IND
GALLIUM ltd.	IND
ELLIOTT s.e.	GBR

我使用了下面的代码 - 当合法名称找到内部键但不使用外部键检查国家/地区条件时，它正在替换字符串。有人可以建议我吗？ （我有一个很大的清单）

for i in range(len(df)):
    for k1 in country_dict.items():
        if df.loc[i, 'Reg Country Code'] == k1:
            for k2, v2 in country_dict[k1].items():
                df.loc[df['Reg Country Code'] == k1, 'Name'] = [re.sub(k2, v, x) if x.endswith(k2) else x for x in df.loc[df['Reg Country Code'] == k1, 'Name']]

我的输出应该是：

Name	Reg Country Code
NexPoint Ltd. Co.	USA
Silverplay PLC	GBR
ALLOYS Pvt. Ltd.	IND
GALLIUM Ltd.	IND
ELLIOTT SE	GBR

Answer 1

您可以按国家代码对df进行分组并替换

df['NAME'] = df.groupby('REG COUNTRY CODE')['NAME'].apply(lambda x: x.replace(d[x.name], regex = True))

    NAME                REG COUNTRY CODE
0   NexPoint Ltd. Co.   USA
1   Silverplay PLC      GBR
2   ALLOYS Pvt. Ltd.    IND
3   GALLIUM Ltd..       IND
4   ELLIOTT SE          GBR

解释：

• df.groupby('REG COUNTRY CODE').name 返回组的名称（在这种情况下为国家代码）。

• 通过使用 d[x.name]，我们可以访问对应字典键（国家代码）的值字典

• 将正则表达式设置为 True 有助于我们部分替换字符串