IIUC，您可以用途：

import re

pat = fr'\b({"|".join(map(re.escape, state_lists))})\b'

df["STATE"] = df["ADDRESS"].str.extract(pat, expand=False)

在你的领域里似乎有int。在应用split()函数之前，可以添加一个条件语句来检查ADDRESS中的值是否是string的示例。您可以将其实现为：

import numpy as np

def extract_state(address):
    if isinstance(address, str):
        state = address.split()[-1]
        return state if state in state_lists else np.nan
    else:
        return np.nan

df["STATE"] = df["ADDRESS"].map(extract_state)