用选定的列表值填写 dataframe 中的列

Question

我在 excel 文件中有很多地址。 我已将其导入并存储在 dataframe 中。 现在我想检测每个地址中的状态并将其显示在新列中。 如何循环我的 dataframe 中的每一行并添加该行中的状态值？

所有状态列表：

allstates=['SELANGOR','JOHOR','KELANTAN','MALACCA','NEGERI SEMBILAN','PAHANG','PENANG','PERAK','PERLIS',
          'SABAH','SARAWAK','TERENGGANU','KUALA LUMPUR','K. LUMPUR','LABUAN','PUTRAJAYA']

以下是我希望我的 dataframe 的样子：

Address                             |    States
-------------------------------------------------------
311 Jalan Springhill SELANGOR       |    *SELANGOR*
31 Jalan Segamat JOHOR              |    *JOHOR*

我希望将状态（例如：SELANGOR）插入状态列

Answer 1

尝试这个：

df['States'] = df.Address.str.extract('({})'.format('|'.join(allstates)))

如果您确定（或希望）state 名称仅出现在地址的末尾：

df['Sates'] = df.Address.str.extract('({})$'.format('|'.join(allstates)))

Output：

                         Address     Sates
0  311 Jalan Springhill SELANGOR  SELANGOR
1         31 Jalan Segamat JOHOR     JOHOR

Answer 2

import pandas as pd
data = pd.read_csv('states.csv')
print(data)
                         Address
0  311 Jalan Springhill SELANGOR
1         31 Jalan Segamat JOHOR

for index, row in data.iterrows():
    value = row.Address
    State = value.split()[-1:][0]
    data.loc[index,'State'] = State

print(data)
                         Address     State
0  311 Jalan Springhill SELANGOR  SELANGOR
1         31 Jalan Segamat JOHOR     JOHOR