Python中使用的模式匹配错误Pandas.series.str。包含字符串替换项

Question

我试图用从包含多组括号的字典中提取的字符串替换pandas数据帧中的字符串。运行脚本时，匹配组出现错误，字符串未被替换。我很有信心这个错误是由括号引起的。

为了解决这个问题，我一直在尝试使用正则表达式模式匹配结构包含（）方法。我已经查看了stackoverflow上提供的其他解决方案，但没有成功地解决我的错误。

下面是一些我用来测试的脚本。在字符串中保留括号很重要（即，我不必删除它们）：

import pandas as pd
import numpy as np

dict= {'2017() (pat)':'2000',
       '2018() (pat)':'2001'}

df = pd.DataFrame({'YEAR': ['test2017end','test2018end','test2019end'],
                   'MONTH': ['Jan','Feb','Mar'],
                   'DD': ['1','12','22']})

for init, repl in dict.items():
    df.loc[df['YEAR'].str.contains(init),'YEAR'] = repl

print(df)

有人能提供关于使用模式匹配的指导，以便正确地替换字符串吗？

谢谢！

Answer 1

不要使用变量dict，因为python代码关键字。在

解决方案是提取字典键中的第一个整数：

import re

d= {'2017() (pat)':'2000',
       '2018() (pat)':'2001'}

df = pd.DataFrame({'YEAR': ['test2017end','test2018end','test2019end'],
                   'MONTH': ['Jan','Feb','Mar'],
                   'DD': ['1','12','22']})

for init, repl in d.items():
    i = re.findall('\d+', init)[0]
    df.loc[df['YEAR'].str.contains(i),'YEAR'] = repl

print(df)
          YEAR MONTH  DD
0         2000   Jan   1
1         2001   Feb  12
2  test2019end   Mar  22