python - 使用选定的列表值填充数据框中的列

Question

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

所有州列表:

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

下面是我想要的数据框:

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

我希望将州(例如:SELANGOR)插入到州列中

Answer 1

试试这个:

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

如果您确定(或希望)州名称仅出现在地址末尾:

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

输出:

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