输出UK地址的不同变体的Python或R函数

import re

def generate_uk_address_variants(address):
    variants = []
    
    # 移除多余的空格
    address = re.sub(r'\s+', ' ', address)
    
    # 分割地址
    parts = address.split(',')
    
    # 提取邮编
    postcode = re.search(r'[A-Z]{1,2}\d{1,2} ?\d[A-Z]{2}', parts[-1])
    
    if postcode:
        postcode = postcode.group(0)
        postcode_parts = postcode.split(' ')
        
        # 移除邮编
        parts[-1] = parts[-1].replace(postcode, '')
        
        # 移除多余的空格
        parts[-1] = parts[-1].strip()
        
        # 生成不同变体的地址
        variants.append(', '.join(parts + [postcode]))
        variants.append(', '.join(parts + [postcode_parts[0], postcode_parts[1]]))
        variants.append(', '.join(parts + [postcode_parts[0] + postcode_parts[1]]))
    
    return variants

address = "123 Main Street, London, UK, SW1A 1AA"
variants = generate_uk_address_variants(address)
print(variants)

['123 Main Street, London, UK, SW1A 1AA', '123 Main Street, London, UK, SW1A 1', '123 Main Street, London, UK, SW1A1']

generate_uk_address_variants <- function(address) {
  variants <- c()
  
  # 移除多余的空格
  address <- gsub("\\s+", " ", address)
  
  # 分割地址
  parts <- strsplit(address, ",")[[1]]
  
  # 提取邮编
  postcode <- regmatches(parts[length(parts)], regexpr("[A-Z]{1,2}\\d{1,2} ?\\d[A-Z]{2}", parts[length(parts)]))
  
  if (length(postcode) > 0) {
    postcode <- postcode[[1]]
    postcode_parts <- strsplit(postcode, " ")[[1]]
    
    # 移除邮编
    parts[length(parts)] <- gsub(postcode, "", parts[length(parts)])
    
    # 移除多余的空格
    parts[length(parts)] <- trimws(parts[length(parts)])
    
    # 生成不同变体的地址
    variants <- c(paste(c(parts, postcode), collapse = ", "),
                  paste(c(parts, postcode_parts[1], postcode_parts[2]), collapse = ", "),
                  paste(c(parts, paste0(postcode_parts[1], postcode_parts[2])), collapse = ", "))
  }
  
  return(variants)
}

address <- "123 Main Street, London, UK, SW1A 1AA"
variants <- generate_uk_address_variants(address)
print(variants)

[1] "123 Main Street, London, UK, SW1A 1AA" "123 Main Street, London, UK, SW1A 1"   "123 Main Street, London, UK, SW1A1"