在地址中搜索数字和字母数字邮政编码

Question

我有一个看起来像这样的熊猫数据框：

Name       Address
Alan       23 Belby road, home near me 71234
Tom        PA23 6NH brickby avenue
Solty      7 solty road 7123-234
Ben        Nowhere road 713456 Belgium

我想获得带有如下结果数据框的邮政编码：

Name       Address                               Postcode
Alan       23 Belby road, home near me 71234     71234
Tom        PA23 6NH brickby avenue               PA23 6NH
Solty      7 solty road 7123-234                 7123-234
Ben        Nowhere road 713456 Belgium           713456

我看了Python, Regular Expression Postcode search和python - get zipcode from full address的帖子

不清楚如何进行。

Answer 1

每个捕获组都可以在由|分隔的re表达式中指定

Extract all 模式匹配到各个列（请参阅。Multiple Pattern using Regex in Pandas）

然后尝试使用 bfill 将所有匹配项强制到第一列（参见。How to collapse multiple columns into one in pandas）

然后合并回原来的数据集。

import pandas as pd

postcode_re = r'([Gg][Ii][Rr] 0[Aa]{2})|' \
              r'((([A-Za-z][0-9]{1,2})|' \
              r'(([A-Za-z][A-Ha-hJ-Yj-y][0-9]{1,2})|' \
              r'(([A-Za-z][0-9][A-Za-z])|' \
              r'([A-Za-z][A-Ha-hJ-Yj-y][0-9][A-Za-z]?))))\s?[0-9][A-Za-z]{2})|' \
              r'(\d{5}\-?\d{0,4})|' \
              r'(\d{4}\-?\d{0,3})'

df = pd.DataFrame({'Name': {0: 'Alan', 1: 'Tom',
                            2: 'Solty', 3: 'Ben',
                            4: 'Mary', 5: 'Mike'},
                   'Address': {0: 'PA23 6NH brickby avenue',
                               1: '818 mention 560100',
                               2: 'calculate AB24 EFT',
                               3: '818 where 560100',
                               4: 'Nowhere road 713456 Belgium',
                               5: '7 solty road 7123-234'}})

df = df.merge(df['Address']
              .str
              .extractall(postcode_re)
              .bfill(axis=1)[0]
              .droplevel(level=1)
              .rename('Postcode'),
              left_index=True,
              right_index=True,
              how='left')

print(df.to_string())

输出：

    Name                      Address  Postcode
0   Alan      PA23 6NH brickby avenue  PA23 6NH
1    Tom           818 mention 560100    560100
2  Solty           calculate AB24 EFT       NaN
3    Ben             818 where 560100    560100
4   Mary  Nowhere road 713456 Belgium    713456
5   Mike        7 solty road 7123-234  7123-234

如果您需要更多邮政编码正则表达式，请参阅postal-codes。